feat: support parameter updates in initialize_dae!

SciML · Aug 6, 2024 · 60a45fa · 60a45fa
1 parent 0428052
commit 60a45fa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/initialize_dae.jl b/src/initialize_dae.jl
@@ -164,6 +164,9 @@ function _initialize_dae!(integrator, prob::Union{ODEProblem, DAEProblem},
     else
         error("Unreachable reached. Report this error.")
     end
+    if isdefined(prob.f, :initializeprobpmap) && prob.f.initializeprobpmap !== nothing
+        integrator.p = prob.f.initializeprobpmap(prob, nlsol)
+    end
 
     if nlsol.retcode != ReturnCode.Success
         integrator.sol = SciMLBase.solution_new_retcode(integrator.sol,