Fix for Optimistix 0.1.0

patrick-kidger · patrick-kidger · commit 47e9a0cc9a57 · 2026-02-17T12:31:50.000+01:00
diff --git 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(
-            jnp.invert(small) & (diverged | jnp.invert(converged)),
+            at_least_two & jnp.invert(small) & (diverged | jnp.invert(converged)),
             optx.RESULTS.nonlinear_divergence,
             optx.RESULTS.successful,
         )

Original file line number	Diff line number	Diff line change
`@@ -162,7 +162,7 @@ def terminate(`
`162`	`162`	`converged = _converged(factor, self.kappa)`
`163`	`163`	`terminate = at_least_two & (small \| diverged \| converged)`
`164`	`164`	`terminate_result = optx.RESULTS.where(`
`165`		`- jnp.invert(small) & (diverged \| jnp.invert(converged)),`
	`165`	`+ at_least_two & jnp.invert(small) & (diverged \| jnp.invert(converged)),`
`166`	`166`	`optx.RESULTS.nonlinear_divergence,`
`167`	`167`	`optx.RESULTS.successful,`
`168`	`168`	`)`