diff --git a/tests/py/dynamo/conversion/test_arange_aten.py b/tests/py/dynamo/conversion/test_arange_aten.py index 32a243330f..1e1e9b3cc7 100644 --- a/tests/py/dynamo/conversion/test_arange_aten.py +++ b/tests/py/dynamo/conversion/test_arange_aten.py @@ -54,6 +54,7 @@ def forward(self, end_tensor): Arange(), inputs, use_example_tensors=False, + check_dtype=False, pyt_inputs=[pyt_input], )