Make non-strict zip strict in scalar/loop.py

Armavica · Armavica · commit bf1e5d97406c · 2024-06-29T03:24:48.000-04:00
diff --git a/pytensor/scalar/loop.py b/pytensor/scalar/loop.py
@@ -93,7 +93,7 @@ def _validate_updates(
                 )
         else:
             update = outputs
-        for i, u in zip(init, update, strict=False):
+        for i, u in zip(init[: len(update)], update, strict=True):
             if i.type != u.type:
                 raise TypeError(
                     "Init and update types must be the same: "

Original file line number	Diff line number	Diff line change
`@@ -93,7 +93,7 @@ def _validate_updates(`
`93`	`93`	`)`
`94`	`94`	`else:`
`95`	`95`	`update = outputs`
`96`		`- for i, u in zip(init, update, strict=False):`
	`96`	`+ for i, u in zip(init[: len(update)], update, strict=True):`
`97`	`97`	`if i.type != u.type:`
`98`	`98`	`raise TypeError(`
`99`	`99`	`"Init and update types must be the same: "`