Skip to content
Snippets Groups Projects
Commit 46d38ae3 authored by auphelia's avatar auphelia
Browse files

[Test] higher precision for isclose comparision of produced and expected output

parent 239c8521
No related branches found
No related tags found
No related merge requests found
......@@ -93,6 +93,6 @@ def test_brevitas_avg_pool_export(
odict = oxe.execute_onnx(model, idict, True)
produced = odict[model.graph.output[0].name]
assert np.isclose(expected, produced, rtol=1e-3).all()
assert np.isclose(expected, produced).all()
os.remove(export_onnx_path)
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