Skip to content

Commit ea7e547

Browse files
committed
Make zips strict in pytensor/sparse
1 parent 78c83e0 commit ea7e547

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytensor/sparse/basic.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2849,7 +2849,7 @@ def choose(continuous, derivative):
28492849
else:
28502850
return None
28512851

2852-
return [choose(c, d) for c, d in zip(is_continuous, derivative)]
2852+
return [choose(c, d) for c, d in zip(is_continuous, derivative, strict=True)]
28532853

28542854
def infer_shape(self, fgraph, node, ins_shapes):
28552855
def _get(l):
@@ -2928,7 +2928,7 @@ def choose(continuous, derivative):
29282928
else:
29292929
return None
29302930

2931-
return [choose(c, d) for c, d in zip(is_continuous, derivative)]
2931+
return [choose(c, d) for c, d in zip(is_continuous, derivative, strict=True)]
29322932

29332933
def infer_shape(self, fgraph, node, ins_shapes):
29342934
def _get(l):

0 commit comments

Comments
 (0)