fix: param bug in test_binary_ops_aten (#2733)

pytorch · May 24, 2024 · 0346a29 · 0346a29
1 parent 3493c12
commit 0346a29
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/py/dynamo/conversion/test_binary_ops_aten.py b/tests/py/dynamo/conversion/test_binary_ops_aten.py
@@ -116,7 +116,7 @@ def forward(self, x):
         inputs = [torch.randn(2, 2)]
         self.run_test(m, inputs)
 
-    @parameterized.expand([((lambda x, y: torch.ops.aten.div.Tensor(x, y)))])
+    @parameterized.expand([(lambda x, y: torch.ops.aten.div.Tensor(x, y),)])
     def test_elementwise_op_div_with_two_ints(self, orig_op: Callable):
         class TestModule(nn.Module):
             def __init__(self, orig_op):
@@ -130,7 +130,7 @@ def forward(self, x):
         inputs = [torch.randint(1, 10, (5,), dtype=torch.int32)]
         self.run_test(m, inputs)
 
-    @parameterized.expand([(lambda x, y: torch.ops.aten.div.Tensor(x, y))])
+    @parameterized.expand([(lambda x, y: torch.ops.aten.div.Tensor(x, y),)])
     def test_elementwise_op_div_with_one_int_one_constant(self, orig_op: Callable):
         class TestModule(nn.Module):
             def __init__(self, orig_op):