From 4c215b4949208e24afb513d412636ffb4fe78574 Mon Sep 17 00:00:00 2001
From: Aayush Sabharwal <aayush.sabharwal@gmail.com>
Date: Mon, 2 Dec 2024 17:56:31 +0530
Subject: [PATCH] fixup! refactor: use `initialization_data` in SciMLFunction
 constructors

---
 src/systems/nonlinear/initializesystem.jl | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/src/systems/nonlinear/initializesystem.jl b/src/systems/nonlinear/initializesystem.jl
index bc9f8386e2..385afc9c5f 100644
--- 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
 
 """