Skip to content

GitLab

Commit 80e63410 authored by Jennifer's avatar Jennifer
Browse files

change message for test_model.py compare

parent 523adaf4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_model.py
View file @ 80e63410
...@@ -202,4 +202,4 @@ class TestModel(unittest.TestCase): ...@@ -202,4 +202,4 @@ class TestModel(unittest.TestCase):
out_repro = out_repro["sm"]["positions"][-1] out_repro = out_repro["sm"]["positions"][-1]
out_repro = out_repro.squeeze(0) 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)
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