Make non-strict zip strict in tensor/subtensor.py

pymc-devs · Jul 23, 2024 · e6069da · e6069da
1 parent 0de617f
commit e6069da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytensor/tensor/subtensor.py b/pytensor/tensor/subtensor.py
@@ -523,7 +523,7 @@ def basic_shape(shape, indices):
 
     """
     res_shape = ()
-    for idx, n in zip(indices, shape, strict=False):
+    for n, idx in zip(shape[: len(indices)], indices, strict=True):
         if isinstance(idx, slice):
             res_shape += (slice_len(idx, n),)
         elif isinstance(getattr(idx, "type", None), SliceType):