Also include u0map in known unknowns; merge keys in a way that actual…

…ly works
SciML · Jul 19, 2024 · 4d62d11 · 4d62d11
1 parent 536aff6
commit 4d62d11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/systems/diffeqs/abstractodesystem.jl b/src/systems/diffeqs/abstractodesystem.jl
@@ -1567,7 +1567,7 @@ function InitializationProblem{iip, specialize}(sys::AbstractODESystem,
     defs = defaults(isys)
 
     # Check that all unknowns have guesses
-    unknowns_missing = setdiff(unknowns(isys), [keys(guesses), keys(defs)])
+    unknowns_missing = setdiff(unknowns(isys), union(keys(guesses), keys(defs), keys(u0map)))
     !isempty(unknowns_missing) &&
         throw(ArgumentError("Missing guesses for unknowns $(join(unknowns_missing, ", "))."))