Make non-strict zip strict in local_subtensor_of_alloc

Author: Armavica

pytensor/tensor/rewriting/subtensor.py

@@ -683,7 +683,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: