Skip to content

Commit b64a49b

Browse files
committed
update batch invariance test assert
1 parent 81e2d4d commit b64a49b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integration/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ def test_batch_invariance(self, fake_model_training_session):
181181
# do not include maxabserr or total energy in testing (per atom energy tested)
182182
if ("maxabserr" in name) or ("total_energy" in name):
183183
continue
184-
if not math.isclose(batch5_val, batch1_val, rel_tol=tol):
184+
if not math.isclose(batch5_val, batch1_val, rel_tol=tol, abs_tol=tol):
185185
raise AssertionError(
186186
f"Validation metric mismatch for '{name}': "
187187
f"batch_size=5 value={batch5_val}, batch_size=1 value={batch1_val}, "

0 commit comments

Comments
 (0)