feat: support SciMLBase.remake_initializeprob

Author: AayushSabharwal

src/systems/nonlinear/initializesystem.jl

@@ -141,3 +141,25 @@ function generate_initializesystem(sys::ODESystem;
 
     return sys_nl
 end
+
+function SciMLBase.remake_initializeprob(sys::ODESystem, odefn, u0, t0, p)
+    if (u0 === missing || !(eltype(u0) <: Pair) || isempty(u0)) &&
+       (p === missing || !(eltype(p) <: Pair) || isempty(p))
+        return odefn.initializeprob, odefn.initializeprobmap, odefn.initializeprobpmap
+    end
+    if u0 === missing
+        u0 = Dict()
+    end
+    if p === missing
+        p = Dict()
+    end
+    u0 = todict(u0)
+    p = todict(p)
+    initprob = InitializationProblem(sys, t0, u0, p)
+    initprobmap = getu(initprob, unknowns(sys))
+    punknowns = [p for p in all_variable_symbols(initprob) if is_parameter(sys, p)]
+    getpunknowns = getu(initprob, punknowns)
+    setpunknowns = setp_oop(sys, punknowns)
+    initprobpmap = GetUpdatedMTKParameters(getpunknowns, setpunknowns)
+    return initprob, initprobmap, initprobpmap
+end