Skip to content

Commit

Permalink
Merge pull request #591 from AayushSabharwal/as/symbolic-save-idxs
Browse files Browse the repository at this point in the history
feat: add support for symbolic `save_idxs`
  • Loading branch information
ChrisRackauckas authored Nov 18, 2024
2 parents 89e62d2 + 725b890 commit 894405c
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
2 changes: 1 addition & 1 deletion Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
5 changes: 3 additions & 2 deletions src/solve.jl
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 894405c

Please sign in to comment.