Skip to content
Snippets Groups Projects
Commit c2e272b9 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

use higher tolerance on np.isclose on ONNX exec test

parent 7d6bfce5
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment