feat: add support for symbolic save_idxs

SciML · Nov 11, 2024 · 4971ffd · 4971ffd
1 parent 03d64e2
commit 4971ffd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/solve.jl b/src/solve.jl
@@ -149,6 +149,8 @@ function DiffEqBase.__init(prob::DiffEqBase.AbstractDDEProblem,
     f_with_history = ODEFunctionWrapper(f, history)
 
     # initialize output arrays of the solution
+    save_idxs, saved_subsystem = SciMLBase.get_save_idxs_and_saved_subsystem(prob, save_idxs)
+
     k = typeof(rate_prototype)[]
     ts, timeseries, ks = solution_arrays(u, tspan, rate_prototype;
         timeseries_init = timeseries_init,
@@ -173,7 +175,7 @@ function DiffEqBase.__init(prob::DiffEqBase.AbstractDDEProblem,
     id = OrdinaryDiffEqCore.InterpolationData(f_with_history, timeseries, ts, ks,
         alg_choice, dense, cache, differential_vars, false)
     sol = DiffEqBase.build_solution(prob, alg.alg, ts, timeseries;
-        dense = dense, k = ks, interp = id,
+        dense = dense, k = ks, interp = id, saved_subsytem = saved_subsystem,
         alg_choice = id.alg_choice, calculate_error = false,
         stats = stats)