Skip to content

Commit 0346a29

Browse files
zewenli98laikhtewari
authored andcommitted
fix: param bug in test_binary_ops_aten (#2733)
1 parent 3493c12 commit 0346a29

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/py/dynamo/conversion/test_binary_ops_aten.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def forward(self, x):
116116
inputs = [torch.randn(2, 2)]
117117
self.run_test(m, inputs)
118118

119-
@parameterized.expand([((lambda x, y: torch.ops.aten.div.Tensor(x, y)))])
119+
@parameterized.expand([(lambda x, y: torch.ops.aten.div.Tensor(x, y),)])
120120
def test_elementwise_op_div_with_two_ints(self, orig_op: Callable):
121121
class TestModule(nn.Module):
122122
def __init__(self, orig_op):
@@ -130,7 +130,7 @@ def forward(self, x):
130130
inputs = [torch.randint(1, 10, (5,), dtype=torch.int32)]
131131
self.run_test(m, inputs)
132132

133-
@parameterized.expand([(lambda x, y: torch.ops.aten.div.Tensor(x, y))])
133+
@parameterized.expand([(lambda x, y: torch.ops.aten.div.Tensor(x, y),)])
134134
def test_elementwise_op_div_with_one_int_one_constant(self, orig_op: Callable):
135135
class TestModule(nn.Module):
136136
def __init__(self, orig_op):

0 commit comments

Comments
 (0)