Skip to content

Commit 4d1019b

Browse files
committed
Remove unnecessary branch and just iterate over empty dictionary
1 parent 7936f98 commit 4d1019b

File tree

1 file changed

+14
-16
lines changed

1 file changed

+14
-16
lines changed

src/systems/nonlinear/initializesystem.jl

Lines changed: 14 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -36,22 +36,20 @@ function generate_initializesystem(sys::ODESystem;
3636
# set dummy derivatives to default_dd_guess unless specified
3737
push!(defs, x[1] => get(guesses, x[1], default_dd_guess))
3838
end
39-
if !isnothing(u0map)
40-
for (y, x) in u0map
41-
y = get(schedule.dummy_sub, y, y)
42-
y = fixpoint_sub(y, diffmap)
43-
if y vars_set
44-
# variables specified in u0 overrides defaults
45-
push!(defs, y => x)
46-
elseif y isa Symbolics.Arr
47-
# TODO: don't scalarize arrays
48-
merge!(defs, Dict(scalarize(y .=> x)))
49-
elseif y isa Symbolics.BasicSymbolic
50-
# y is a derivative expression expanded; add it to the initialization equations
51-
push!(eqs_ics, y ~ x)
52-
else
53-
error("Initialization expression $y is currently not supported. If its a higher order derivative expression, then only the dummy derivative expressions are supported.")
54-
end
39+
for (y, x) in u0map
40+
y = get(schedule.dummy_sub, y, y)
41+
y = fixpoint_sub(y, diffmap)
42+
if y vars_set
43+
# variables specified in u0 overrides defaults
44+
push!(defs, y => x)
45+
elseif y isa Symbolics.Arr
46+
# TODO: don't scalarize arrays
47+
merge!(defs, Dict(scalarize(y .=> x)))
48+
elseif y isa Symbolics.BasicSymbolic
49+
# y is a derivative expression expanded; add it to the initialization equations
50+
push!(eqs_ics, y ~ x)
51+
else
52+
error("Initialization expression $y is currently not supported. If its a higher order derivative expression, then only the dummy derivative expressions are supported.")
5553
end
5654
end
5755
end

0 commit comments

Comments
 (0)