chore: fix arange test

pytorch · Aug 22, 2024 · 0dcc3ed · 0dcc3ed
1 parent db15eca
commit 0dcc3ed
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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],
         )