@@ -14,7 +14,7 @@ function initializesystem(sys::ODESystem; name = nameof(sys), guesses = Dict(),
14
14
u0 = Vector {Pair} (undef, 0 )
15
15
defs = ModelingToolkit. defaults (sys)
16
16
17
- full_states = [sts;getfield .((observed (sys)),:lhs )]
17
+ full_states = [sts; getfield .((observed (sys)), :lhs )]
18
18
19
19
# Refactor to ODESystem construction
20
20
# should be ModelingToolkit.guesses(sys)
@@ -28,26 +28,28 @@ function initializesystem(sys::ODESystem; name = nameof(sys), guesses = Dict(),
28
28
def = defs[st]
29
29
30
30
if def isa Equation
31
- st ∉ keys (guesses) && error (" Invalid setup: unknown $(st) has an initial condition equation with no guess." )
32
- push! (eqs_ics,def)
33
- push! (u0,st => guesses[st])
31
+ st ∉ keys (guesses) &&
32
+ error (" Invalid setup: unknown $(st) has an initial condition equation with no guess." )
33
+ push! (eqs_ics, def)
34
+ push! (u0, st => guesses[st])
34
35
else
35
- push! (eqs_ics,st ~ def)
36
+ push! (eqs_ics, st ~ def)
36
37
push! (u0, st => def)
37
38
end
38
39
elseif st ∈ keys (guesses)
39
- push! (u0,st => guesses[st])
40
+ push! (u0, st => guesses[st])
40
41
else
41
42
error (" Invalid setup: unknown $(st) has no default value or initial guess" )
42
43
end
43
44
end
44
45
45
46
pars = parameters (sys)
47
+ nleqs = [eqs_ics; observed (sys)]
46
48
47
- sys_nl = NonlinearSystem ([eqs_ics; observed (sys)] ,
49
+ sys_nl = NonlinearSystem (nleqs ,
48
50
full_states,
49
51
pars;
50
- defaults = merge (ModelingToolkit. defaults (sys),todict (u0)),
52
+ defaults = merge (ModelingToolkit. defaults (sys), todict (u0)),
51
53
name,
52
54
kwargs... )
53
55
0 commit comments