diff --git a/tests/py/dynamo/conversion/test_local_scalar_dense_aten.py b/tests/py/dynamo/conversion/test_local_scalar_dense_aten.py index 4d22632789..3576c5c9fa 100644 --- a/tests/py/dynamo/conversion/test_local_scalar_dense_aten.py +++ b/tests/py/dynamo/conversion/test_local_scalar_dense_aten.py @@ -8,24 +8,12 @@ class TestLocalScalarDenseConverter(DispatchTestCase): @parameterized.expand( [ - ( - torch.randn((5, 10, 5), dtype=torch.float32), - ), - ( - torch.randint(-10, 10, (5, 1, 15), dtype=torch.int32), - ), - ( - torch.randn((1), dtype=torch.float32), - ), - ( - (torch.tensor([-2.4])), - ), - ( - (torch.tensor([5.5, 3.5, 3.6])), - ), - ( - (torch.tensor([True])), - ), + (torch.randn((5, 10, 5), dtype=torch.float32),), + (torch.randint(-10, 10, (5, 1, 15), dtype=torch.int32),), + (torch.randn((1), dtype=torch.float32),), + ((torch.tensor([-2.4])),), + ((torch.tensor([5.5, 3.5, 3.6])),), + ((torch.tensor([True])),), ( torch.tensor( [ @@ -44,9 +32,7 @@ class TestLocalScalarDenseConverter(DispatchTestCase): ] ), ), - ( - (torch.tensor([float("inf")])), - ), + ((torch.tensor([float("inf")])),), ] ) def test_local_scalar_dense(self, data):