From 46d38ae370bf3bae67982bcc531442f1d4afc245 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Tue, 9 Jun 2020 14:54:41 +0100
Subject: [PATCH] [Test] higher precision for isclose comparision of produced
and expected output
---
tests/brevitas/test_brevitas_avg_pool_export.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/brevitas/test_brevitas_avg_pool_export.py b/tests/brevitas/test_brevitas_avg_pool_export.py
index a423b89ff..3e90e9830 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)
--
GitLab