added bug fixed for init dt0

patrick-kidger · Jul 28, 2023 · b71f8fa · b71f8fa
1 parent 2638a35
commit b71f8fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/diffrax/integrate.py b/diffrax/integrate.py
@@ -598,7 +598,7 @@ def diffeqsolve(
         )
         with jax.ensure_compile_time_eval():
             pred = (t1 - t0) * dt0 < 0
-        dt0 = eqxi.error_if(dt0, pred, msg)
+        dt0 = eqxi.error_if(jnp.array(dt0), pred, msg)
 
     # Backward compatibility
     if isinstance(