diff --git a/diffrax/_root_finder/_verychord.py b/diffrax/_root_finder/_verychord.py index d73b3336..f58fcb44 100644 --- a/diffrax/_root_finder/_verychord.py +++ b/diffrax/_root_finder/_verychord.py @@ -162,7 +162,7 @@ def terminate( converged = _converged(factor, self.kappa) terminate = at_least_two & (small | diverged | converged) terminate_result = optx.RESULTS.where( - at_least_two & jnp.invert(small) & (diverged | jnp.invert(converged)), + at_least_two & jnp.invert(small) & diverged, optx.RESULTS.nonlinear_divergence, optx.RESULTS.successful, ) diff --git a/test/test_solver.py b/test/test_solver.py index a022f644..dfff3f6b 100644 --- a/test/test_solver.py +++ b/test/test_solver.py @@ -50,7 +50,7 @@ def test_implicit_euler_adaptive(): stepsize_controller=stepsize_controller, throw=False, ) - assert out1.result == diffrax.RESULTS.nonlinear_divergence + assert out1.result == diffrax.RESULTS.nonlinear_max_steps_reached assert out2.result == diffrax.RESULTS.successful