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)