Skip to content

Commit cc5bc2f

Browse files
committed
relax test_avgpool rtol
1 parent b15f792 commit cc5bc2f

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/test_backend.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -190,8 +190,6 @@ def test_maxpool(self):
190190
self.log.debug(str(p))
191191
self._run_test_case([_OUTPUT], {_INPUT: x_val})
192192

193-
@unittest.skipIf(get_test_config().is_onnxruntime_backend and get_test_config().backend_version == "0.2.1"
194-
and (not get_test_config().is_mac), "onnxruntime bug")
195193
@check_onnxruntime_incompatibility("AveragePool")
196194
def test_avgpool(self):
197195
for tf_shape in ["known", "unknown"]:
@@ -207,7 +205,7 @@ def test_avgpool(self):
207205
_ = tf.identity(mp, name=_TFOUTPUT)
208206

209207
self.log.debug(str(p))
210-
self._run_test_case([_OUTPUT], {_INPUT: x_val})
208+
self._run_test_case([_OUTPUT], {_INPUT: x_val}, rtol=1e-06)
211209

212210
def _conv_test(self, x_val, w, strides=None, padding="VALID", dilations=None, rtol=1e-07):
213211
if strides is None:

0 commit comments

Comments
 (0)