diff --git a/tests/lbfgs_test.py b/tests/lbfgs_test.py index 12b49a40..c9fc9d93 100644 --- a/tests/lbfgs_test.py +++ b/tests/lbfgs_test.py @@ -419,7 +419,7 @@ def binary_logit_log_likelihood_jax(beta, y, x): onp.asarray(y), onp.asarray(x)) jaxopt_val = binary_logit_log_likelihood(jaxopt_res, y, x) - self.assertArraysAllClose(scipy_val, jaxopt_val) + self.assertArraysAllClose(scipy_val, jaxopt_val, rtol=3e-5) @parameterized.product(linesearch=['zoom', 'backtracking', 'hager-zhang'])