No public description

PiperOrigin-RevId: 653533491
google · Jul 18, 2024 · 6d88292 · 6d88292
1 parent 4a50198
commit 6d88292
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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'])