minor edit

vroulet · Jun 7, 2023 · 8da43d3 · 8da43d3
1 parent 416e687
commit 8da43d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/zoom_linesearch2_test.py b/tests/zoom_linesearch2_test.py
@@ -291,7 +291,7 @@ def fun(x):
     ls = ZoomLineSearch(fun)
     _, state = ls.run(init_stepsize=1.0, params=x, descent_direction=p)
 
-    assert not state.failed
+    self.assertFalse(state.failed) 
 
   @parameterized.product(out_dtype=[jnp.float32, jnp.float64])
   def test_correct_dtypes(self, out_dtype):