Scalarize initial conditions of vectors

SciML · Oct 3, 2024 · e9a06c5 · e9a06c5
1 parent 5cf19a9
commit e9a06c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/systems/nonlinear/initializesystem.jl b/src/systems/nonlinear/initializesystem.jl
@@ -45,7 +45,7 @@ function generate_initializesystem(sys::ODESystem;
                     push!(defs, y => x)
                 elseif y isa Symbolics.Arr
                     # TODO: don't scalarize arrays
-                    push!(defs, (collect(y) .=> x)...)
+                    merge!(defs, Dict(scalarize(y .=> x)))
                 elseif y isa Symbolics.BasicSymbolic
                     # y is a derivative expression expanded; add it to the initialization equations
                     push!(eqs_ics, y ~ x)