diff --git a/tests/test_basic_onnx_exec.py b/tests/test_basic_onnx_exec.py index b47412c0bfaf5d7d54095c691968aae989a35de3..0bc05628a5a498476332b397833a35056d08acfa 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)