Make zip strict

pymc-devs · Jun 21, 2024 · 9d81ab4 · 9d81ab4
1 parent a14cb2b
commit 9d81ab4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytensor/tensor/random/utils.py b/pytensor/tensor/random/utils.py
@@ -129,7 +129,8 @@ def explicit_expand_dims(
     """Introduce explicit expand_dims in RV parameters that are implicitly broadcasted together and/or by size."""
 
     batch_dims = [
-        param.type.ndim - ndim_param for param, ndim_param in zip(params, ndim_params)
+        param.type.ndim - ndim_param
+        for param, ndim_param in zip(params, ndim_params, strict=True)
     ]
 
     if size_length is not None: