mirror of
https://github.com/aqlaboratory/openfold.git
synced 2026-06-04 12:44:26 +08:00
change message for test_model.py compare
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user