diff --git a/source/tests/pt/test_padding_atoms.py b/source/tests/pt/test_padding_atoms.py index cad81f107d..8fb417c6a4 100644 --- a/source/tests/pt/test_padding_atoms.py +++ b/source/tests/pt/test_padding_atoms.py @@ -49,7 +49,8 @@ def setUp(self) -> None: self.sel = [16, 8] self.rcut = 2.2 self.rcut_smth = 0.4 - self.atol = 1e-8 + self.atol = 1e-6 + self.rtol = 1e-5 class TestPaddingAtoms(unittest.TestCase, TestCaseSingleFrameWithoutNlist): @@ -79,6 +80,7 @@ def test_padding_atoms_consistency(self): to_numpy_array(result[var_name].cpu().detach()), np.mean(to_numpy_array(result[f"atom_{var_name}"].cpu().detach()), axis=1), atol=self.atol, + rtol=self.rtol, ) # test padding atoms padding_atoms_list = [1, 5, 10] @@ -105,6 +107,7 @@ def test_padding_atoms_consistency(self): to_numpy_array(result[var_name].cpu().detach()), to_numpy_array(result_padding[var_name].cpu().detach()), atol=self.atol, + rtol=self.rtol, )