Skip to content

Commit 0dcc3ed

Browse files
committed
chore: fix arange test
1 parent db15eca commit 0dcc3ed

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/py/dynamo/conversion/test_arange_aten.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@ def forward(self, end_tensor):
5454
Arange(),
5555
inputs,
5656
use_example_tensors=False,
57+
check_dtype=False,
5758
pyt_inputs=[pyt_input],
5859
)
5960

0 commit comments

Comments
 (0)