change message for test_model.py compare

80e63410 · Jennifer · 523adaf4 · 80e63410
Commit 80e63410 authored Apr 17, 2024 by Jennifer
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)