Merge pull request #610 from ArnoStrouwen/enzyme

make sensitivity_solution work with Enzyme
SciML · Feb 4, 2024 · 0daefce · 0daefce
2 parents 0c62711 + a165c34
commit 0daefce
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/solutions/ode_solutions.jl b/src/solutions/ode_solutions.jl
@@ -399,9 +399,9 @@ function sensitivity_solution(sol::ODESolution, u, t)
     end
 
     interp = enable_interpolation_sensitivitymode(sol.interp)
-
-    ODESolution{T, N}(u, sol.u_analytic, sol.errors, t,
-        nothing, sol.prob,
+    ODESolution{T, N}(u, sol.u_analytic, sol.errors, 
+        t isa Vector ? t : collect(t),
+        sol.k, sol.prob,
         sol.alg, interp,
         sol.dense, sol.tslocation,
         sol.stats, sol.alg_choice, sol.retcode)