fixup! refactor: use initialization_data in SciMLFunction constructors

SciML · Dec 2, 2024 · 4c215b4 · 4c215b4
1 parent 2beb33a
commit 4c215b4
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/src/systems/nonlinear/initializesystem.jl b/src/systems/nonlinear/initializesystem.jl
@@ -329,13 +329,7 @@ function SciMLBase.remake_initialization_data(sys::ODESystem, odefn, u0, t0, p,
     filter_missing_values!(pmap)
     f, _ = process_SciMLProblem(EmptySciMLFunction, sys, u0map, pmap; guesses, t = t0)
     kws = f.kwargs
-    initprob = get(kws, :initializeprob, nothing)
-    if initprob === nothing
-        return nothing
-    end
-    return SciMLBase.OverrideInitData(initprob, get(kws, :update_initializeprob!, nothing),
-        get(kws, :initializeprobmap, nothing),
-        get(kws, :initializeprobpmap, nothing))
+    return get(kws, :initialization_data, nothing)
 end
 
 """