Skip to content

Commit 8286e1b

Browse files
author
Ian Schweer
committed
Allow skip tests
1 parent 913ea6b commit 8286e1b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/link/pytorch/test_blockwise.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,11 @@
55
import pytensor.tensor as pt
66
from pytensor.graph.basic import Apply
77
from pytensor.graph.op import Op
8-
from pytensor.link.pytorch.dispatch.basic import pytorch_funcify
98
from pytensor.tensor.blockwise import Blockwise
109

1110

1211
torch = pytest.importorskip("torch")
12+
basic = pytest.importorskip("pytensor.link.pytorch.dispatch.basic")
1313

1414

1515
class TestOp(Op):
@@ -27,7 +27,7 @@ def perform(self, *_):
2727
raise RuntimeError("In perform")
2828

2929

30-
@pytorch_funcify.register(TestOp)
30+
@basic.pytorch_funcify.register(TestOp)
3131
def evaluate_test_op(op, **_):
3232
@torch.compiler.disable(recursive=False)
3333
def func(a, b):

0 commit comments

Comments
 (0)