Skip to content

Commit 39075ea

Browse files
committed
made_jump=True for both solver steps
1 parent 2fbd6ee commit 39075ea

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

diffrax/_solver/reversible.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def step(
8888
y1 = (self.l * (ω(y0) - ω(z0)) + ω(step_z0)).ω
8989

9090
step_y1, y_error, _, _, result2 = self.solver.step(
91-
terms, t1, t0, y1, args, original_solver_state, False
91+
terms, t1, t0, y1, args, original_solver_state, True
9292
)
9393
z1 = (ω(y1) + ω(z0) - ω(step_y1)).ω
9494

@@ -113,7 +113,7 @@ def backward_step(
113113
)
114114
z0 = (ω(z1) - ω(y1) + ω(step_y1)).ω
115115
step_z0, _, dense_info, _, _ = self.solver.step(
116-
terms, t0, t1, z0, args, original_solver_state, False
116+
terms, t0, t1, z0, args, original_solver_state, True
117117
)
118118
y0 = ((1 / self.l) * (ω(y1) - ω(step_z0)) + ω(z0)).ω
119119
solver_state = (original_solver_state, z0)

0 commit comments

Comments
 (0)