From 8831462f3bc964c0daa4425dcbcc0635d1470959 Mon Sep 17 00:00:00 2001 From: Hoonkyung Cho Date: Thu, 13 Jun 2024 08:27:10 +0900 Subject: [PATCH] chore: small linting --- .../test_local_scalar_dense_aten.py | 28 +++++-------------- 1 file changed, 7 insertions(+), 21 deletions(-) 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):