Make zips strict in pytensor/misc

Armavica · Armavica · commit 829498017d33 · 2024-06-24T10:35:44.000-04:00
diff --git a/pytensor/misc/check_blas.py b/pytensor/misc/check_blas.py
@@ -59,7 +59,7 @@ def execute(execute=True, verbose=True, M=2000, N=2000, K=2000, iters=10, order=
     if any(x.op.__class__.__name__ == "Gemm" for x in f.maker.fgraph.toposort()):
         c_impl = [
             hasattr(thunk, "cthunk")
-            for node, thunk in zip(f.vm.nodes, f.vm.thunks)
+            for node, thunk in zip(f.vm.nodes, f.vm.thunks, strict=True)
             if node.op.__class__.__name__ == "Gemm"
         ]
         assert len(c_impl) == 1

Original file line number	Diff line number	Diff line change
`@@ -59,7 +59,7 @@ def execute(execute=True, verbose=True, M=2000, N=2000, K=2000, iters=10, order=`
`59`	`59`	`if any(x.op.__class__.__name__ == "Gemm" for x in f.maker.fgraph.toposort()):`
`60`	`60`	`c_impl = [`
`61`	`61`	`hasattr(thunk, "cthunk")`
`62`		`- for node, thunk in zip(f.vm.nodes, f.vm.thunks)`
	`62`	`+ for node, thunk in zip(f.vm.nodes, f.vm.thunks, strict=True)`
`63`	`63`	`if node.op.__class__.__name__ == "Gemm"`
`64`	`64`	`]`
`65`	`65`	`assert len(c_impl) == 1`