Make zip strict

ricardoV94 · ricardoV94 · commit 9d81ab493a6d · 2024-06-21T14:05:11.000+02:00
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:

Original file line number	Diff line number	Diff line change
`@@ -129,7 +129,8 @@ def explicit_expand_dims(`
`129`	`129`	`"""Introduce explicit expand_dims in RV parameters that are implicitly broadcasted together and/or by size."""`
`130`	`130`
`131`	`131`	`batch_dims = [`
`132`		`- param.type.ndim - ndim_param for param, ndim_param in zip(params, ndim_params)`
	`132`	`+ param.type.ndim - ndim_param`
	`133`	`+ for param, ndim_param in zip(params, ndim_params, strict=True)`
`133`	`134`	`]`
`134`	`135`
`135`	`136`	`if size_length is not None:`