TypeError in DecomposeMultiDimSqueeze TVM callback in densenet xray variant

[kamalrajkannan78]

Description

• TypeError: int() argument must be a string, a bytes-like object or a real number, not 'Any' in DecomposeMultiDimSqueeze TVM callback in densenet xray variant

Reproduce

git checkout kkannan/model_issues
git submodule update --recursive
pytest forge/test/model_demos/high_prio/cnn/pytorch/test_densenet.py -k "xray" -svv

Observed Behaviour

>   input_shape = [int(dim) for dim in pre.args[0].checked_type.shape]
E   TypeError: int() argument must be a string, a bytes-like object or a real number, not 'Any'

Logs

• densenet_xray.log