Skip to content

Commit 36eb83f

Browse files
fix: fix conversion of solved array parameters to variables in initialization
1 parent 7abe134 commit 36eb83f

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/systems/nonlinear/initializesystem.jl

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -174,6 +174,14 @@ function generate_initializesystem(sys::ODESystem;
174174
defs[eq.lhs] = eq.rhs
175175
end
176176

177+
# even if `p => tovar(p)` is in `paramsubs`, `isparameter(p[1]) === true` after substitution
178+
# so add scalarized versions as well
179+
for k in collect(keys(paramsubs))
180+
symbolic_type(k) == ArraySymbolic() || continue
181+
for i in eachindex(k)
182+
paramsubs[k[i]] = paramsubs[k][i]
183+
end
184+
end
177185
eqs_ics = Symbolics.substitute.([eqs_ics; trueobs], (paramsubs,))
178186
vars = [vars; collect(values(paramsubs))]
179187
for k in keys(defs)

0 commit comments

Comments
 (0)