Skip to content

Commit 25e0f29

Browse files
authored
Merge pull request #370 from nbcsm/test_fix
Test fix
2 parents 20dae5d + c48eee9 commit 25e0f29

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tests/test_backend.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -190,8 +190,8 @@ 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")
193+
@unittest.skipIf(get_test_config().is_onnxruntime_backend and get_test_config().backend_version == "0.2.1",
194+
"onnxruntime bug")
195195
@check_onnxruntime_incompatibility("AveragePool")
196196
def test_avgpool(self):
197197
for tf_shape in ["known", "unknown"]:
@@ -207,7 +207,7 @@ def test_avgpool(self):
207207
_ = tf.identity(mp, name=_TFOUTPUT)
208208

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

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

tests/test_lstm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -335,7 +335,7 @@ def test_dynamic_basiclstm(self):
335335
feed_dict = {"input_1:0": x_val}
336336
input_names_with_port = ["input_1:0"]
337337
output_names_with_port = ["output:0", "cell_state:0"]
338-
self.run_test_case(feed_dict, input_names_with_port, output_names_with_port, rtol=0.0001)
338+
self.run_test_case(feed_dict, input_names_with_port, output_names_with_port, rtol=0.0001, atol=1e-06)
339339

340340
def test_dynamic_lstm_output_consumed_only(self):
341341
units = 5

0 commit comments

Comments
 (0)