Skip to content

Commit d0b9fc2

Browse files
committed
Disable caching of to_tuple helper due to failure in Python 3.10
1 parent ad15c54 commit d0b9fc2

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pytensor/link/numba/dispatch/blockwise.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,9 @@ def numba_funcify_Blockwise(op: BlockwiseWithCoreShape, node, **kwargs):
6060
src,
6161
"to_tuple",
6262
global_env={"to_fixed_tuple": to_fixed_tuple},
63-
)
63+
),
64+
# cache=True leads to a numba.cloudpickle dump failure in Python 3.10
65+
cache=False,
6466
)
6567

6668
def blockwise_wrapper(*inputs_and_core_shapes):

0 commit comments

Comments
 (0)