Skip to content

Commit f1c186c

Browse files
committed
bump equivariance testing fp64 tol for nequip models
1 parent 8039d02 commit f1c186c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/unit/model/test_nequip_model.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,10 @@ class TestNequIPModel(TorchSimIntegrationMixin, LAMMPSMLIAPIntegrationMixin):
8080
def strict_locality(self):
8181
return False
8282

83+
@pytest.fixture(scope="class")
84+
def equivariance_tol(self, model_dtype):
85+
return {"float32": 5e-5, "float64": 1e-7}[model_dtype]
86+
8387
@pytest.fixture(
8488
params=[
8589
minimal_config1,

0 commit comments

Comments
 (0)