Allow skip tests

pymc-devs · Oct 30, 2024 · 8286e1b · 8286e1b
1 parent 913ea6b
commit 8286e1b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/link/pytorch/test_blockwise.py b/tests/link/pytorch/test_blockwise.py
@@ -5,11 +5,11 @@
 import pytensor.tensor as pt
 from pytensor.graph.basic import Apply
 from pytensor.graph.op import Op
-from pytensor.link.pytorch.dispatch.basic import pytorch_funcify
 from pytensor.tensor.blockwise import Blockwise
 
 
 torch = pytest.importorskip("torch")
+basic = pytest.importorskip("pytensor.link.pytorch.dispatch.basic")
 
 
 class TestOp(Op):
@@ -27,7 +27,7 @@ def perform(self, *_):
         raise RuntimeError("In perform")
 
 
-@pytorch_funcify.register(TestOp)
+@basic.pytorch_funcify.register(TestOp)
 def evaluate_test_op(op, **_):
     @torch.compiler.disable(recursive=False)
     def func(a, b):