From c2e272b98e08cef5ec4c1f223bc882afd975e2c1 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 15 Oct 2019 23:06:36 +0100
Subject: [PATCH] use higher tolerance on np.isclose on ONNX exec test
---
tests/test_basic_onnx_exec.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/tests/test_basic_onnx_exec.py b/tests/test_basic_onnx_exec.py
index b47412c0b..0bc05628a 100644
--- a/tests/test_basic_onnx_exec.py
+++ b/tests/test_basic_onnx_exec.py
@@ -32,7 +32,8 @@ def test_mnist_onnx_download_extract_run():
input_dict = {"Input3": np_helper.to_array(input_tensor)}
output_dict = oxe.execute_onnx(model, input_dict)
assert np.isclose(
- np_helper.to_array(output_tensor), output_dict["Plus214_Output_0"]
+ np_helper.to_array(output_tensor), output_dict["Plus214_Output_0"],
+ atol=1e-3
).all()
# remove the downloaded model and extracted files
os.remove(dl_ret)
--
GitLab