Fix OPM test

c2f46ce8 · Gustaf Ahdritz · 07f98811 · c2f46ce8
Commit c2f46ce8 authored Jun 22, 2022 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_outer_product_mean.py tests/test_outer_product_mean.py +1 -1

No files found.
--- a/tests/test_outer_product_mean.py
+++ b/tests/test_outer_product_mean.py
@@ -92,7 +92,7 @@ class TestOuterProductMean(unittest.TestCase):

        # Even when correct, OPM has large, precision-related errors. It gets
        # a special pass from consts.eps.
-        self.assertTrue(torch.max(torch.abs(out_gt - out_repro) < 5e-4))
+        self.assertTrue(torch.max(torch.abs(out_gt - out_repro)) < 5e-4)


 if __name__ == "__main__":