Skip to content

Commit c752dac

Browse files
committed
fix: arrays_for_pooling and test_torch_avg_pool2d
1 parent 88583d4 commit c752dac

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

ivy_tests/test_ivy/helpers/hypothesis_helpers/array_helpers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1810,6 +1810,7 @@ def arrays_for_pooling(
18101810
num_arrays=1,
18111811
max_value=100,
18121812
min_value=-100,
1813+
abs_smallest_val=1e-04,
18131814
)
18141815
)
18151816

ivy_tests/test_ivy/test_frontends/test_torch/test_nn/test_functional/test_pooling_functions.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -292,6 +292,8 @@ def test_torch_avg_pool2d(
292292
ceil_mode=ceil_mode,
293293
count_include_pad=count_include_pad,
294294
divisor_override=None,
295+
atol=1e-2,
296+
rtol=1e-2,
295297
)
296298

297299

0 commit comments

Comments
 (0)