Internal changes.

PiperOrigin-RevId: 675234113
google · Sep 16, 2024 · eb969a1 · eb969a1
1 parent 6d88292
commit eb969a1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/lbfgs_test.py b/tests/lbfgs_test.py
@@ -419,8 +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, rtol=3e-5)
-
+    self.assertLessEqual(jaxopt_val, scipy_val + 3e-5)
 
   @parameterized.product(linesearch=['zoom', 'backtracking', 'hager-zhang'])
   def test_complex(self, linesearch):