Skip to content

Commit fe7018c

Browse files
committed
Make non-strict zip strict in tensor/subtensor.py
1 parent 267e32c commit fe7018c

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
@@ -522,7 +522,7 @@ def basic_shape(shape, indices):
522522
523523
"""
524524
res_shape = ()
525-
for idx, n in zip(indices, shape, strict=False):
525+
for n, idx in zip(shape[: len(indices)], indices, strict=True):
526526
if isinstance(idx, slice):
527527
res_shape += (slice_len(idx, n),)
528528
elif isinstance(getattr(idx, "type", None), SliceType):

0 commit comments

Comments
 (0)