Merge pull request #3136 from AayushSabharwal/as/array-guesses

fix: handle array guesses in `InitializationProblem`
SciML · Oct 22, 2024 · d96793b · d96793b
2 parents 74412cd + f531800
commit d96793b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -126,7 +126,7 @@ SpecialFunctions = "0.7, 0.8, 0.9, 0.10, 1.0, 2"
 StaticArrays = "0.10, 0.11, 0.12, 1.0"
 SymbolicIndexingInterface = "0.3.31"
 SymbolicUtils = "3.7"
-Symbolics = "6.14"
+Symbolics = "6.15.2"
 URIs = "1"
 UnPack = "0.1, 1.0"
 Unitful = "1.1"

diff --git a/src/systems/diffeqs/abstractodesystem.jl b/src/systems/diffeqs/abstractodesystem.jl
@@ -1311,7 +1311,11 @@ function InitializationProblem{iip, specialize}(sys::AbstractODESystem,
         u0T = promote_type(u0T, typeof(fullmap[eq.lhs]))
     end
     if u0T != Union{}
-        u0map = Dict(k => symbolic_type(v) == NotSymbolic() ? u0T(v) : v
+        u0map = Dict(k => if symbolic_type(v) == NotSymbolic() && !is_array_of_symbolics(v)
+                         v isa AbstractArray ? u0T.(v) : u0T(v)
+                     else
+                         v
+                     end
         for (k, v) in u0map)
     end
     if neqs == nunknown