Fix check

Xilinx · Nov 15, 2023 · 328d3a6 · 328d3a6
1 parent 5b8ef5f
commit 328d3a6
Showing 1 changed file with 3 additions and 4 deletions.
diff --git 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)