don't specialize integrate on df (#423)

qojulia · Oct 17, 2024 · 1cff06f · 1cff06f
1 parent 8fe86a0
commit 1cff06f
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/src/timeevolution_base.jl b/src/timeevolution_base.jl
@@ -11,18 +11,20 @@ function recast! end
 
 Integrate using OrdinaryDiffEq
 """
-function integrate(tspan, df::F, x0,
+function integrate(tspan, df, x0,
             state, dstate, fout;
             alg = OrdinaryDiffEq.DP5(),
             steady_state = false, tol = 1e-3, save_everystep = false, saveat=tspan,
-            callback = nothing, kwargs...) where {F}
-
-    function df_(dx, x, p, t)
-        recast!(state,x)
-        recast!(dstate,dx)
-        df(t, state, dstate)
-        recast!(dx,dstate)
-        return nothing
+            callback = nothing, kwargs...)
+
+    df_ = let df = df
+        function df_(dx, x, p, t)
+            recast!(state,x)
+            recast!(dstate,dx)
+            df(t, state, dstate)
+            recast!(dx,dstate)
+            return nothing
+        end
     end
 
     fout_ = let fout = fout, state = state