From cc307e1bb994b57228724f0a3ccaabcb88638c51 Mon Sep 17 00:00:00 2001 From: CosmoMatt Date: Fri, 14 Jun 2024 16:24:28 +0100 Subject: [PATCH] reduce gradient tests tolerance --- tests/test_gradients.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_gradients.py b/tests/test_gradients.py index 035ee41..abae6c9 100644 --- a/tests/test_gradients.py +++ b/tests/test_gradients.py @@ -57,4 +57,4 @@ def func(flm): loss += jnp.mean(jnp.abs(coeffs[i])) return loss - check_grads(func, (flm,), order=1, modes=("rev"), rtol=5e-3) + check_grads(func, (flm,), order=1, modes=("rev"), rtol=1e-2)