diff --git a/tests/brevitas/test_brevitas_avg_pool_export.py b/tests/brevitas/test_brevitas_avg_pool_export.py index a423b89ff95eb6b43290312b4531e689b51d1c36..3e90e9830900daaf42daf7996f945a268ba0f23b 100644 --- a/tests/brevitas/test_brevitas_avg_pool_export.py +++ b/tests/brevitas/test_brevitas_avg_pool_export.py @@ -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)