From 328d3a6d0c7f8c14c22f4f2fac38327f13f9d7ec Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Tue, 14 Nov 2023 16:47:37 +0000 Subject: [PATCH] Fix check --- tests/brevitas_ort/common.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/brevitas_ort/common.py b/tests/brevitas_ort/common.py index 2249f2bd0..c05fd59b9 100644 --- a/tests/brevitas_ort/common.py +++ b/tests/brevitas_ort/common.py @@ -146,10 +146,9 @@ def is_brevitas_ort_close( ort_output = ort_output[0] if isinstance(brevitas_output, tuple): brevitas_output = brevitas_output[0] - - # make sure we are not comparing 0s - if (ort_output == 0).all() and (brevitas_output == 0).all(): - pytest.skip("Skip testing against all 0s.") + # make sure we are not comparing 0s + if (ort_output == 0).all() and (brevitas_output == 0).all(): + pytest.skip("Skip testing against all 0s.") return recursive_allclose(ort_output, brevitas_output, tolerance)