Make tests pass on aarch64-linux (#671)

blackjax-devs · May 13, 2024 · af79fa4 · af79fa4
1 parent 40efb6c
commit af79fa4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/adaptation/test_adaptation.py b/tests/adaptation/test_adaptation.py
@@ -66,6 +66,6 @@ def test_chees_adaptation():
     )(chain_keys, last_states)
 
     harmonic_mean = 1.0 / jnp.mean(1.0 / infos.acceptance_rate)
-    np.testing.assert_allclose(harmonic_mean, 0.75, rtol=1e-1)
+    np.testing.assert_allclose(harmonic_mean, 0.75, atol=1e-1)
     np.testing.assert_allclose(parameters["step_size"], 1.5, rtol=2e-1)
     np.testing.assert_array_less(infos.num_integration_steps.mean(), 15.0)