change message for test_model.py compare

cdf60396 · Jennifer · Jennifer Wei · 19c81588 · cdf60396
Commit cdf60396 authored Apr 17, 2024 by Jennifer Committed by Jennifer Wei May 06, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_model.py tests/test_model.py +1 -1

No files found.
--- a/tests/test_model.py
+++ b/tests/test_model.py
@@ -202,4 +202,4 @@ class TestModel(unittest.TestCase):
        out_repro = out_repro["sm"]["positions"][-1]
        out_repro = out_repro.squeeze(0)

-        self.assertTrue(torch.max(torch.abs(out_gt - out_repro)) < 1e-3)
+        compare_utils.assert_mean_abs_diff_small(out_gt, out_repro, 1e-3)