chore: Update test_SKHamiltonian to use torch.allclose with specified…

… tolerances
deepmodeling · Sep 24, 2024 · c48612a · c48612a
1 parent cced890
commit c48612a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dptb/tests/test_SKHamiltonian.py b/dptb/tests/test_SKHamiltonian.py
@@ -222,6 +222,6 @@ def test_onsite_stain(self):
           0.0000000000e+00,  2.6077032089e-08,  0.0000000000e+00,
          -3.7709136009e+00]])
 
-        assert torch.allclose(data[AtomicDataDict.NODE_FEATURES_KEY], expected_strainonsite)
+        assert torch.allclose(data[AtomicDataDict.NODE_FEATURES_KEY], expected_strainonsite, atol=1e-6, rtol=1e-4)