fix(numba): Remove useless blockwise in tests

pymc-devs · Jun 11, 2024 · dbe0e09 · dbe0e09
1 parent d92c367
commit dbe0e09
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/link/numba/test_basic.py b/tests/link/numba/test_basic.py
@@ -118,8 +118,12 @@ def perform(self, node, inputs, outputs):
 my_multi_out.ufunc = MyMultiOut.impl
 my_multi_out.ufunc.nin = 2
 my_multi_out.ufunc.nout = 2
-opts = RewriteDatabaseQuery(include=[None], exclude=["cxx_only", "BlasOpt"])
-numba_mode = Mode(NumbaLinker(), opts.including("numba"))
+opts = RewriteDatabaseQuery(
+    include=[None], exclude=["cxx_only", "BlasOpt", "local_careduce_fusion"]
+)
+numba_mode = Mode(
+    NumbaLinker(), opts.including("numba", "local_useless_unbatched_blockwise")
+)
 py_mode = Mode("py", opts)
 
 rng = np.random.default_rng(42849)