Skip to content

Commit

Permalink
Make non-strict zip strict in local_subtensor_of_alloc
Browse files Browse the repository at this point in the history
  • Loading branch information
Armavica committed Jul 23, 2024
1 parent 6b39026 commit 0de617f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytensor/tensor/rewriting/subtensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -648,7 +648,7 @@ def local_subtensor_of_alloc(fgraph, node):
# Slices to take from val
val_slices = []

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

0 comments on commit 0de617f

Please sign in to comment.