From 8c2e978a4a09e56de5d128e21f8b5c0bd27dbea7 Mon Sep 17 00:00:00 2001 From: Aayush Sabharwal Date: Sun, 23 Oct 2022 18:13:37 +0530 Subject: [PATCH] Refactor to `u0map` --- src/systems/optimization/optimizationsystem.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/systems/optimization/optimizationsystem.jl b/src/systems/optimization/optimizationsystem.jl index bf68584a97..26b31cdb93 100644 --- a/src/systems/optimization/optimizationsystem.jl +++ b/src/systems/optimization/optimizationsystem.jl @@ -329,7 +329,7 @@ function OptimizationProblemExpr(sys::OptimizationSystem, args...; kwargs...) OptimizationProblemExpr{true}(sys::OptimizationSystem, args...; kwargs...) end -function OptimizationProblemExpr{iip}(sys::OptimizationSystem, u0, +function OptimizationProblemExpr{iip}(sys::OptimizationSystem, u0map, parammap = DiffEqBase.NullParameters(); lb = nothing, ub = nothing, grad = false, @@ -366,7 +366,7 @@ function OptimizationProblemExpr{iip}(sys::OptimizationSystem, u0, defs = defaults(sys) defs = mergedefaults(defs, parammap, ps) - defs = mergedefaults(defs, u0, dvs) + defs = mergedefaults(defs, u0map, dvs) u0 = varmap_to_vars(u0map, dvs; defaults = defs, tofloat = false) p = varmap_to_vars(parammap, ps; defaults = defs, tofloat = false, use_union)