From 0346a29c8d997dd3aa38d0f92ef6afc4b75361f5 Mon Sep 17 00:00:00 2001 From: "Zewen (Evan) Li" Date: Wed, 17 Apr 2024 04:11:56 +0800 Subject: [PATCH] fix: param bug in `test_binary_ops_aten` (#2733) --- tests/py/dynamo/conversion/test_binary_ops_aten.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/py/dynamo/conversion/test_binary_ops_aten.py b/tests/py/dynamo/conversion/test_binary_ops_aten.py index 331fab591d..ebe727b716 100644 --- 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):