From 0dcc3ed3cd61d54fab08db3f572d746164994cfa Mon Sep 17 00:00:00 2001
From: Dheeraj Peri <peri.dheeraj@gmail.com>
Date: Thu, 22 Aug 2024 00:30:07 -0700
Subject: [PATCH] chore: fix arange test

---
 tests/py/dynamo/conversion/test_arange_aten.py | 1 +
 1 file changed, 1 insertion(+)

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],
         )