Skip to content

GitLab

Commit c2f46ce8 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Fix OPM test

parent 07f98811
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_outer_product_mean.py
View file @ c2f46ce8
...@@ -92,7 +92,7 @@ class TestOuterProductMean(unittest.TestCase): ...@@ -92,7 +92,7 @@ class TestOuterProductMean(unittest.TestCase):
# Even when correct, OPM has large, precision-related errors. It gets # Even when correct, OPM has large, precision-related errors. It gets
# a special pass from consts.eps. # 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__": if __name__ == "__main__":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment