Skip to content

Commit 9248437

Browse files
Merge pull request #982 from AayushSabharwal/as/remake-typeinf
fix: fix type inference in `remake`
2 parents b5288ad + f7e9798 commit 9248437

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

src/remake.jl

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@ function remake(prob::ODEProblem; f = missing,
217217
p = missing,
218218
kwargs = missing,
219219
interpret_symbolicmap = true,
220-
build_initializeprob = true,
220+
build_initializeprob = Val{true},
221221
use_defaults = false,
222222
lazy_initialization = nothing,
223223
_kwargs...)
@@ -229,7 +229,7 @@ function remake(prob::ODEProblem; f = missing,
229229

230230
iip = isinplace(prob)
231231

232-
if build_initializeprob
232+
if build_initializeprob <: Val{true} || build_initializeprob == true
233233
if f !== missing && has_initialization_data(f)
234234
initialization_data = remake_initialization_data(
235235
prob.f.sys, f, u0, tspan[1], p, newu0, newp)
@@ -401,15 +401,15 @@ function remake(prob::SDEProblem;
401401
seed = missing,
402402
kwargs = missing,
403403
lazy_initialization = nothing,
404-
build_initializeprob = true,
404+
build_initializeprob = Val{true},
405405
_kwargs...)
406406
if tspan === missing
407407
tspan = prob.tspan
408408
end
409409

410410
newu0, newp = updated_u0_p(prob, u0, p, tspan[1]; interpret_symbolicmap, use_defaults)
411411

412-
if build_initializeprob
412+
if build_initializeprob <: Val{true} || build_initializeprob == true
413413
if f !== missing && has_initialization_data(f)
414414
initialization_data = remake_initialization_data(
415415
prob.f.sys, f, u0, tspan[1], p, newu0, newp)
@@ -462,15 +462,15 @@ function remake(prob::DDEProblem; f = missing, h = missing, u0 = missing,
462462
tspan = missing, p = missing, constant_lags = missing,
463463
dependent_lags = missing, order_discontinuity_t0 = missing,
464464
neutral = missing, kwargs = missing, interpret_symbolicmap = true,
465-
use_defaults = false, lazy_initialization = nothing, build_initializeprob = true,
465+
use_defaults = false, lazy_initialization = nothing, build_initializeprob = Val{true},
466466
_kwargs...)
467467
if tspan === missing
468468
tspan = prob.tspan
469469
end
470470

471471
newu0, newp = updated_u0_p(prob, u0, p, tspan[1]; interpret_symbolicmap, use_defaults)
472472

473-
if build_initializeprob
473+
if build_initializeprob <: Val{true} || build_initializeprob == true
474474
if f !== missing && has_initialization_data(f)
475475
initialization_data = remake_initialization_data(
476476
prob.f.sys, f, u0, tspan[1], p, newu0, newp)
@@ -535,15 +535,15 @@ function remake(prob::SDDEProblem;
535535
seed = missing,
536536
kwargs = missing,
537537
lazy_initialization = nothing,
538-
build_initializeprob = true,
538+
build_initializeprob = Val{true},
539539
_kwargs...)
540540
if tspan === missing
541541
tspan = prob.tspan
542542
end
543543

544544
newu0, newp = updated_u0_p(prob, u0, p, tspan[1]; interpret_symbolicmap, use_defaults)
545545

546-
if build_initializeprob
546+
if build_initializeprob <: Val{true} || build_initializeprob == true
547547
if f !== missing && has_initialization_data(f)
548548
initialization_data = remake_initialization_data(
549549
prob.f.sys, f, u0, tspan[1], p, newu0, newp)
@@ -681,11 +681,11 @@ function remake(prob::NonlinearProblem;
681681
interpret_symbolicmap = true,
682682
use_defaults = false,
683683
lazy_initialization = nothing,
684-
build_initializeprob = true,
684+
build_initializeprob = Val{true},
685685
_kwargs...)
686686
newu0, newp = updated_u0_p(prob, u0, p; interpret_symbolicmap, use_defaults)
687687

688-
if build_initializeprob
688+
if build_initializeprob <: Val{true} || build_initializeprob == true
689689
if f !== missing && has_initialization_data(f)
690690
initialization_data = remake_initialization_data(
691691
prob.f.sys, f, u0, nothing, p, newu0, newp)
@@ -728,11 +728,11 @@ function remake(prob::SteadyStateProblem;
728728
interpret_symbolicmap = true,
729729
use_defaults = false,
730730
lazy_initialization = nothing,
731-
build_initializeprob = true,
731+
build_initializeprob = Val{true},
732732
_kwargs...)
733733
newu0, newp = updated_u0_p(prob, u0, p; interpret_symbolicmap, use_defaults)
734734

735-
if build_initializeprob
735+
if build_initializeprob <: Val{true} || build_initializeprob == true
736736
if f !== missing && has_initialization_data(f)
737737
initialization_data = remake_initialization_data(
738738
prob.f.sys, f, u0, Inf, p, newu0, newp)
@@ -769,10 +769,10 @@ Remake the given `NonlinearLeastSquaresProblem`.
769769
"""
770770
function remake(prob::NonlinearLeastSquaresProblem; f = missing, u0 = missing, p = missing,
771771
interpret_symbolicmap = true, use_defaults = false, kwargs = missing,
772-
lazy_initialization = nothing, build_initializeprob = true, _kwargs...)
772+
lazy_initialization = nothing, build_initializeprob = Val{true}, _kwargs...)
773773
newu0, newp = updated_u0_p(prob, u0, p; interpret_symbolicmap, use_defaults)
774774

775-
if build_initializeprob
775+
if build_initializeprob <: Val{true} || build_initializeprob == true
776776
if f !== missing && has_initialization_data(f)
777777
initialization_data = remake_initialization_data(
778778
prob.f.sys, f, u0, nothing, p, newu0, newp)

0 commit comments

Comments
 (0)