make sensitivity_solution work with Enzyme

SciML · Feb 4, 2024 · 7d9bf90 · 7d9bf90
1 parent 0c62711
commit 7d9bf90
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/solutions/ode_solutions.jl b/src/solutions/ode_solutions.jl
@@ -399,9 +399,8 @@ 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, collect(t),
+        sol.k, sol.prob,
         sol.alg, interp,
         sol.dense, sol.tslocation,
         sol.stats, sol.alg_choice, sol.retcode)