Slightly increase error tolerance

desy-ml · Dec 13, 2024 · 5292a34 · 5292a34
1 parent be050f8
commit 5292a34
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_vectorized.py b/tests/test_vectorized.py
@@ -488,11 +488,11 @@ def test_vectorized_aperture_broadcasting(aperture_shape):
         assert np.allclose(
             outgoing.survival_probabilities.mean(dim=-1)[:, 0],
             [0.077, 0.945, 0.995],
-            atol=1e-3,  # Last digit off by one
+            atol=2e-3,  # Last digit off by two
         )
     elif aperture_shape == "rectangular":
         assert np.allclose(
             outgoing.survival_probabilities.mean(dim=-1)[:, 0],
             [0.079, 0.954, 0.997],
-            atol=1e-3,  # Last digit off by one
+            atol=2e-3,  # Last digit off by two
         )