From aeaaf9cb4917e89d7df209023e606045727551c8 Mon Sep 17 00:00:00 2001
From: Aayush Sabharwal <aayush.sabharwal@gmail.com>
Date: Wed, 6 Nov 2024 17:03:35 +0530
Subject: [PATCH 1/2] feat: add support for symbolic `save_idxs`

---
 src/solve.jl | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/solve.jl b/src/solve.jl
index 28911c46..d7bc2f8c 100644
--- a/src/solve.jl
+++ b/src/solve.jl
@@ -230,6 +230,7 @@ function DiffEqBase.__init(
   d_discontinuities_internal = OrdinaryDiffEq.initialize_d_discontinuities(tType, d_discontinuities, tspan)
 
   ### Algorithm-specific defaults ###
+  save_idxs, saved_subsystem = SciMLBase.get_save_idxs_and_saved_subsystem(prob, save_idxs)
   # if save_idxs === nothing
   #   ksEltype = Vector{rateType}
   # else
@@ -538,12 +539,12 @@ function DiffEqBase.__init(
   if alg isa Union{StochasticDiffEqCompositeAlgorithm,
                           StochasticDiffEqRODECompositeAlgorithm}
     sol = DiffEqBase.build_solution(prob,alg,ts,timeseries,W=W,
-                                    stats = stats,
+                                    stats = stats, saved_subsystem = saved_subsystem,
                                     calculate_error = false, alg_choice=alg_choice,
                                     interp = id, dense = dense, seed = _seed)
   else
     sol = DiffEqBase.build_solution(prob,alg,ts,timeseries,W=W,
-                                    stats = stats,
+                                    stats = stats, saved_subsystem = saved_subsystem,
                                     calculate_error = false,
                                     interp = id, dense = dense, seed = _seed)
   end

From 725b890dce055d8118a298fcbacb02b6518131a4 Mon Sep 17 00:00:00 2001
From: Aayush Sabharwal <aayush.sabharwal@gmail.com>
Date: Mon, 11 Nov 2024 09:26:16 +0530
Subject: [PATCH 2/2] build: bump SciMLBase compat

---
 Project.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Project.toml b/Project.toml
index 752d22ae..630b3d6d 100644
--- a/Project.toml
+++ b/Project.toml
@@ -52,7 +52,7 @@ Random = "1.6"
 RandomNumbers = "1.5.3"
 RecursiveArrayTools = "2, 3"
 Reexport = "0.2, 1.0"
-SciMLBase = "2.51"
+SciMLBase = "2.59.2"
 SciMLOperators = "0.2.9, 0.3"
 SparseArrays = "1.6"
 SparseDiffTools = "2"