Skip to content

Commit

Permalink
Enlarge tolerance of regression check to avoid the error raised from …
Browse files Browse the repository at this point in the history
…numeric calculations.
  • Loading branch information
sovietdevil committed Jul 29, 2024
1 parent b91754f commit a64ea16
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/test_ase.py
Original file line number Diff line number Diff line change
Expand Up @@ -223,13 +223,13 @@ def test_shiftml2_regression_mean():
out_ensemble = model.get_cs_iso_ensemble(frame)

assert np.allclose(
out_mean.flatten(), expected_mean_v2
out_mean.flatten(), expected_mean_v2, atol=0.01
), "ShiftML2 failed regression mean test"

assert np.allclose(
out_std.flatten(), expected_std_v2
out_std.flatten(), expected_std_v2, atol=0.01
), "ShiftML2 failed regression variance test"

assert np.allclose(
out_ensemble.flatten(), expected_ensemble_v2.flatten()
out_ensemble.flatten(), expected_ensemble_v2.flatten(), atol=0.01
), "ShiftML2 failed regression ensemble test"

0 comments on commit a64ea16

Please sign in to comment.