From 19b0e7d3ab8aa810d6a25292e9ca4358ec8ece89 Mon Sep 17 00:00:00 2001 From: Alberto Cabezas Gonzalez Date: Mon, 13 May 2024 15:07:53 +0100 Subject: [PATCH] robustify test_chees_adaptation --- tests/adaptation/test_adaptation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/adaptation/test_adaptation.py b/tests/adaptation/test_adaptation.py index 104c8abb9..f54d18c21 100644 --- 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)