fixup! feat: allow specifying variable names in modelingtoolkitize

Author: AayushSabharwal

src/systems/optimization/modelingtoolkitize.jl

@@ -20,6 +20,13 @@ function modelingtoolkitize(prob::DiffEqBase.OptimizationProblem;
         varidxs = variable_index.((prob.f.sys,), varnames)
         invpermute!(varnames, varidxs)
         _vars = [variable(name) for name in varnames]
+        if prob.f.sys isa OptimizationSystem
+            for (i, sym) in enumerate(variable_symbols(prob.f.sys))
+                if hasbounds(sym)
+                    _vars[i] = Symbolics.setmetadata(_vars[i], VariableBounds, getbounds(sym))
+                end
+            end
+        end
     else
         _vars = [variable(:x, i) for i in eachindex(prob.u0)]
     end
@@ -103,9 +110,24 @@ function modelingtoolkitize(prob::DiffEqBase.OptimizationProblem;
     else
         vec(collect(params))
     end
-    de = OptimizationSystem(eqs, vec(vars), params;
+
+    sts = vec(collect(vars))
+    default_u0 = Dict(sts .=> vec(collect(prob.u0)))
+    default_p = if has_p
+        if prob.p isa AbstractDict
+            Dict(v => prob.p[k] for (k, v) in pairs(_params))
+        elseif prob.p isa MTKParameters
+            Dict(params .=> reduce(vcat, prob.p))
+        else
+            Dict(params .=> vec(collect(prob.p)))
+        end
+    else
+        Dict()
+    end
+    de = OptimizationSystem(eqs, sts, params;
         name = gensym(:MTKizedOpt),
         constraints = cons,
+        defaults = merge(default_u0, default_p),
         kwargs...)
     de
 end