added bug fixed for init dt0

thibmonsel · thibmonsel · commit 3bfa34e4e118 · 2023-07-27T10:08:11.000+02:00
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(

Original file line number	Diff line number	Diff line change
`@@ -598,7 +598,7 @@ def diffeqsolve(`
`598`	`598`	`)`
`599`	`599`	`with jax.ensure_compile_time_eval():`
`600`	`600`	`pred = (t1 - t0) * dt0 < 0`
`601`		`- dt0 = eqxi.error_if(dt0, pred, msg)`
	`601`	`+ dt0 = eqxi.error_if(jnp.array(dt0), pred, msg)`
`602`	`602`
`603`	`603`	`# Backward compatibility`
`604`	`604`	`if isinstance(`