Switch updating time stamp to update with time stamp after predicting (…

…#678)
pnkraemer · Nov 20, 2023 · 41c7d0c · 41c7d0c
1 parent 5a558e5
commit 41c7d0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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