fix: handle non-symbolic and nothing u0 for DiscreteProblem

SciML · Sep 19, 2024 · bbc8bf7 · bbc8bf7
1 parent e855035
commit bbc8bf7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/systems/discrete_system/discrete_system.jl b/src/systems/discrete_system/discrete_system.jl
@@ -247,6 +247,13 @@ function process_DiscreteProblem(constructor, sys::DiscreteSystem, u0map, paramm
     dvs = unknowns(sys)
     ps = parameters(sys)
 
+    if eltype(u0map) <: Number
+        u0map = unknowns(sys) .=> vec(u0map)
+    end
+    if u0map === nothing || isempty(u0map)
+        u0map = Dict()
+    end
+
     trueu0map = Dict()
     for (k, v) in u0map
         k = unwrap(k)