Skip to content

Commit 9c09207

Browse files
refactor: add backward compat wrapper
1 parent e07a234 commit 9c09207

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

src/remake.jl

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ function remake(prob::ODEProblem; f = missing,
125125

126126
if f === missing
127127
if build_initializeprob
128-
initialization_data = remake_initialization_data(
128+
initialization_data = remake_initialization_data_compat_wrapper(
129129
prob.f.sys, prob.f, u0, tspan[1], p, newu0, newp)
130130
else
131131
initialization_data = nothing
@@ -202,6 +202,21 @@ function remake_initializeprob(sys, scimlfn, u0, t0, p)
202202
scimlfn.update_initializeprob!, scimlfn.initializeprobmap, scimlfn.initializeprobpmap
203203
end
204204

205+
"""
206+
$(TYPEDSIGNATURES)
207+
208+
Wrapper around `remake_initialization_data` for backward compatibility when `newu0` and
209+
`newp` were not arguments.
210+
"""
211+
function remake_initialization_data_compat_wrapper(sys, scimlfn, u0, t0, p, newu0, newp)
212+
if hasmethod(remake_initialization_data,
213+
Tuple{typeof(sys), typeof(scimlfn), typeof(u0), typeof(t0), typeof(p)})
214+
remake_initialization_data(sys, scimlfn, u0, t0, p)
215+
else
216+
remake_initialization_data(sys, scimlfn, u0, t0, p, newu0, newp)
217+
end
218+
end
219+
205220
"""
206221
remake_initialization_data(sys, scimlfn, u0, t0, p, newu0, newp)
207222

0 commit comments

Comments
 (0)