Skip to content

Commit

Permalink
chore: small linting
Browse files Browse the repository at this point in the history
  • Loading branch information
chohk88 committed Apr 24, 2024
1 parent d83081e commit 8a80daf
Showing 1 changed file with 7 additions and 21 deletions.
28 changes: 7 additions & 21 deletions tests/py/dynamo/conversion/test_local_scalar_dense_aten.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
[
Expand All @@ -44,9 +32,7 @@ class TestLocalScalarDenseConverter(DispatchTestCase):
]
),
),
(
(torch.tensor([float("inf")])),
),
((torch.tensor([float("inf")])),),
]
)
def test_local_scalar_dense(self, data):
Expand Down

0 comments on commit 8a80daf

Please sign in to comment.