Merge pull request patrick-kidger#68 from patrick-kidger/windows-fix

Fix for patrick-kidger#67
thibmonsel · Feb 15, 2022 · ebee2fe · ebee2fe
2 parents 29372ab + b2cc38a
commit ebee2fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/diffrax/step_size_controller/adaptive.py b/diffrax/step_size_controller/adaptive.py
@@ -467,7 +467,7 @@ def _scale(_y0, _y1_candidate, _y_error):
         # Clip next step size based on dtmin/dtmax
         #
 
-        result = jnp.full_like(t0, RESULTS.successful, dtype=int)
+        result = RESULTS.successful
         if self.dtmax is not None:
             dt = jnp.minimum(dt, self.dtmax)
         if self.dtmin is None: