diff --git a/probdiffeq/solvers/strategies/strategy.py b/probdiffeq/solvers/strategies/strategy.py index f911e0cb..0922a4da 100644 --- a/probdiffeq/solvers/strategies/strategy.py +++ b/probdiffeq/solvers/strategies/strategy.py @@ -100,10 +100,10 @@ def init(self, t, posterior, /) -> _State: def predict_error(self, state: _State, /, *, dt, vector_field): """Predict the error of an upcoming step.""" hidden, extra = self.extrapolation.begin(state.hidden, state.aux_extra, dt=dt) + t = state.t + dt error, observed, corr = self.correction.estimate_error( - hidden, state.aux_corr, vector_field=vector_field, t=state.t + hidden, state.aux_corr, vector_field=vector_field, t=t ) - t = state.t + dt state = _State(t=t, hidden=hidden, aux_extra=extra, aux_corr=corr) return error, observed, state