start root finder at time of previous ode_solver_step than final ode_solver_step

LuggiStruggi · LuggiStruggi · commit 506c5be6b094 · 2025-05-27T11:42:04.000+02:00
diff --git a/diffrax/_integrate.py b/diffrax/_integrate.py
@@ -727,7 +727,7 @@ def _call_real_impl():
             _event_root_find = optx.root_find(
                 _to_root_find,
                 event.root_finder,
-                y0=final_state.event_tnext,
+                y0=final_state.event_tprev,
                 options=_options,
                 throw=False,
             )

Original file line number	Diff line number	Diff line change
`@@ -727,7 +727,7 @@ def _call_real_impl():`
`727`	`727`	`_event_root_find = optx.root_find(`
`728`	`728`	`_to_root_find,`
`729`	`729`	`event.root_finder,`
`730`		`- y0=final_state.event_tnext,`
	`730`	`+ y0=final_state.event_tprev,`
`731`	`731`	`options=_options,`
`732`	`732`	`throw=False,`
`733`	`733`	`)`