Skip to content

Commit fdc80a3

Browse files
committed
Make non-strict zip strict in test_abstract_conv
1 parent c28cc2f commit fdc80a3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/tensor/conv/test_abstract_conv.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1745,7 +1745,7 @@ def setup_method(self):
17451745
self.random_stream = np.random.default_rng(utt.fetch_seed())
17461746

17471747
self.inputs_shapes = [(8, 1, 12, 12), (1, 1, 5, 5), (1, 1, 5, 6), (1, 1, 6, 6)]
1748-
self.filters_shapes = [(5, 1, 2, 2), (1, 1, 3, 3)]
1748+
self.filters_shapes = [(5, 1, 2, 2), (1, 1, 3, 3)] * 2
17491749

17501750
self.subsamples = [(1, 1), (2, 2)]
17511751
self.border_modes = ["valid", "full"]
@@ -1764,7 +1764,7 @@ def test_conv2d_grad_wrt_inputs(self):
17641764
# results are the same.
17651765

17661766
for in_shape, fltr_shape in zip(
1767-
self.inputs_shapes, self.filters_shapes, strict=False
1767+
self.inputs_shapes, self.filters_shapes, strict=True
17681768
):
17691769
for bm in self.border_modes:
17701770
for ss in self.subsamples:
@@ -1828,7 +1828,7 @@ def test_conv2d_grad_wrt_weights(self):
18281828
# results are the same.
18291829

18301830
for in_shape, fltr_shape in zip(
1831-
self.inputs_shapes, self.filters_shapes, strict=False
1831+
self.inputs_shapes, self.filters_shapes, strict=True
18321832
):
18331833
for bm in self.border_modes:
18341834
for ss in self.subsamples:

0 commit comments

Comments
 (0)