Skip to content

Commit ae66e82

Browse files
Ian SchweerricardoV94
Ian Schweer
authored andcommitted
Fix test warning
1 parent 7300a68 commit ae66e82

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/link/pytorch/test_blockwise.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
basic = pytest.importorskip("pytensor.link.pytorch.dispatch.basic")
1313

1414

15-
class TestOp(Op):
15+
class BatchedTestOp(Op):
1616
gufunc_signature = "(m,n),(n,p)->(m,p)"
1717

1818
def __init__(self, final_shape):
@@ -27,7 +27,7 @@ def perform(self, *_):
2727
raise RuntimeError("In perform")
2828

2929

30-
@basic.pytorch_funcify.register(TestOp)
30+
@basic.pytorch_funcify.register(BatchedTestOp)
3131
def evaluate_test_op(op, **_):
3232
def func(a, b):
3333
op.call_shapes.extend(map(torch.Tensor.size, [a, b]))
@@ -42,7 +42,7 @@ def test_blockwise_broadcast():
4242

4343
x = pt.tensor4("x", shape=(5, 1, 2, 3))
4444
y = pt.tensor3("y", shape=(3, 3, 2))
45-
op = TestOp((2, 2))
45+
op = BatchedTestOp((2, 2))
4646
z = Blockwise(op)(x, y)
4747

4848
f = pytensor.function([x, y], z, mode="PYTORCH")

0 commit comments

Comments
 (0)