reduce gradient tests tolerance

astro-informatics · Jun 14, 2024 · cc307e1 · cc307e1
1 parent 9300f08
commit cc307e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)