Skip to content

Commit

Permalink
fixup! fix: fix initialization with defaults dependent on indepvar
Browse files Browse the repository at this point in the history
  • Loading branch information
AayushSabharwal committed Jul 22, 2024
1 parent 2662c60 commit 22487e6
Showing 1 changed file with 14 additions and 1 deletion.
15 changes: 14 additions & 1 deletion src/systems/diffeqs/abstractodesystem.jl
Original file line number Diff line number Diff line change
Expand Up @@ -717,14 +717,19 @@ function get_u0_p(sys,
end
p = varmap_to_vars(parammap, ps; defaults = defs, tofloat, use_union)
p = p === nothing ? SciMLBase.NullParameters() : p
t0 !== nothing && delete!(defs, get_iv(sys))
u0, p, defs
end

function get_u0(
sys, u0map, parammap = nothing; symbolic_u0 = false, toterm = default_toterm)
sys, u0map, parammap = nothing; symbolic_u0 = false,
toterm = default_toterm, t0 = nothing)
dvs = unknowns(sys)
ps = parameters(sys)
defs = defaults(sys)
if t0 !== nothing
defs[get_iv(sys)] = t0
end
if parammap !== nothing
defs = mergedefaults(defs, parammap, ps)
end
Expand All @@ -745,6 +750,7 @@ function get_u0(
else
u0 = varmap_to_vars(u0map, dvs; defaults = defs, tofloat = true, toterm)
end
t0 !== nothing && delete!(defs, get_iv(sys))
return u0, defs
end

Expand Down Expand Up @@ -832,6 +838,12 @@ function process_DEProblem(constructor, sys::AbstractODESystem, u0map, parammap;
all(isequal(Continuous()), ci.var_domain) &&
ModelingToolkit.get_tearing_state(sys) !== nothing) ||
!isempty(initialization_equations(sys))) && t !== nothing
if eltype(u0map) <: Number
u0map = unknowns(sys) .=> u0map
end
if isempty(u0map)
u0map = Dict()
end
initializeprob = ModelingToolkit.InitializationProblem(
sys, t, u0map, parammap; guesses, warn_initialize_determined,
initialization_eqs, eval_expression, eval_module, fully_determined)
Expand Down Expand Up @@ -862,6 +874,7 @@ function process_DEProblem(constructor, sys::AbstractODESystem, u0map, parammap;
parammap;
tofloat,
use_union,
t0 = t,
symbolic_u0)
p, split_idxs = split_parameters_by_type(p)
if p isa Tuple
Expand Down

0 comments on commit 22487e6

Please sign in to comment.