Disable caching of to_tuple helper due to failure in Python 3.10

pymc-devs · Nov 27, 2024 · d0b9fc2 · d0b9fc2
1 parent ad15c54
commit d0b9fc2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pytensor/link/numba/dispatch/blockwise.py b/pytensor/link/numba/dispatch/blockwise.py
@@ -60,7 +60,9 @@ def numba_funcify_Blockwise(op: BlockwiseWithCoreShape, node, **kwargs):
             src,
             "to_tuple",
             global_env={"to_fixed_tuple": to_fixed_tuple},
-        )
+        ),
+        # cache=True leads to a numba.cloudpickle dump failure in Python 3.10
+        cache=False,
     )
 
     def blockwise_wrapper(*inputs_and_core_shapes):