Skip to content

Commit 9e9d65d

Browse files
committed
Make non-strict zip strict in tensor/subtensor.py
1 parent be1f800 commit 9e9d65d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytensor/tensor/subtensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -523,7 +523,7 @@ def basic_shape(shape, indices):
523523
524524
"""
525525
res_shape = ()
526-
for idx, n in zip(indices, shape, strict=False):
526+
for n, idx in zip(shape[: len(indices)], indices, strict=True):
527527
if isinstance(idx, slice):
528528
res_shape += (slice_len(idx, n),)
529529
elif isinstance(getattr(idx, "type", None), SliceType):

0 commit comments

Comments
 (0)