Skip to content

Commit 0de617f

Browse files
committed
Make non-strict zip strict in local_subtensor_of_alloc
1 parent 6b39026 commit 0de617f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytensor/tensor/rewriting/subtensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -648,7 +648,7 @@ def local_subtensor_of_alloc(fgraph, node):
648648
# Slices to take from val
649649
val_slices = []
650650

651-
for i, (sl, dim) in enumerate(zip(slices, dims, strict=False)):
651+
for i, (sl, dim) in enumerate(zip(slices, dims[: len(slices)], strict=True)):
652652
# If val was not copied over that dim,
653653
# we need to take the appropriate subtensor on it.
654654
if i >= n_added_dims:

0 commit comments

Comments
 (0)