Skip to content

Commit 8831462

Browse files
committed
chore: small linting
1 parent 7c69c31 commit 8831462

File tree

1 file changed

+7
-21
lines changed

1 file changed

+7
-21
lines changed

tests/py/dynamo/conversion/test_local_scalar_dense_aten.py

+7-21
Original file line numberDiff line numberDiff line change
@@ -8,24 +8,12 @@
88
class TestLocalScalarDenseConverter(DispatchTestCase):
99
@parameterized.expand(
1010
[
11-
(
12-
torch.randn((5, 10, 5), dtype=torch.float32),
13-
),
14-
(
15-
torch.randint(-10, 10, (5, 1, 15), dtype=torch.int32),
16-
),
17-
(
18-
torch.randn((1), dtype=torch.float32),
19-
),
20-
(
21-
(torch.tensor([-2.4])),
22-
),
23-
(
24-
(torch.tensor([5.5, 3.5, 3.6])),
25-
),
26-
(
27-
(torch.tensor([True])),
28-
),
11+
(torch.randn((5, 10, 5), dtype=torch.float32),),
12+
(torch.randint(-10, 10, (5, 1, 15), dtype=torch.int32),),
13+
(torch.randn((1), dtype=torch.float32),),
14+
((torch.tensor([-2.4])),),
15+
((torch.tensor([5.5, 3.5, 3.6])),),
16+
((torch.tensor([True])),),
2917
(
3018
torch.tensor(
3119
[
@@ -44,9 +32,7 @@ class TestLocalScalarDenseConverter(DispatchTestCase):
4432
]
4533
),
4634
),
47-
(
48-
(torch.tensor([float("inf")])),
49-
),
35+
((torch.tensor([float("inf")])),),
5036
]
5137
)
5238
def test_local_scalar_dense(self, data):

0 commit comments

Comments
 (0)