fix: fix remake for IntervalNonlinearProblem

SciML · Jun 28, 2024 · ea1e723 · ea1e723
1 parent c2fb81f
commit ea1e723
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/remake.jl b/src/remake.jl
@@ -63,6 +63,11 @@ function remake(prob::AbstractSciMLProblem; u0 = missing,
     _remake_internal(prob; kwargs..., u0, p)
 end
 
+function remake(prob::AbstractIntervalNonlinearProblem; p = missing, interpret_symbolicmap = true, use_defaults = false, kwargs...)
+    _, p = updated_u0_p(prob, [], p; interpret_symbolicmap, use_defaults)
+    _remake_internal(prob; kwargs..., p)
+end
+
 function remake(prob::AbstractNoiseProblem; kwargs...)
     _remake_internal(prob; kwargs...)
 end

diff --git a/test/remake_tests.jl b/test/remake_tests.jl
@@ -265,3 +265,11 @@ u0 = [1.0; 2.0; 3.0]
 tspan = (0.0, 100.0)
 prob = ODEProblem(lorenz!, u0, tspan, nothing)
 @test_nowarn remake(prob, p = (), interpret_symbolicmap = true)
+
+# IntervalNonlinearProblem doesn't have a u0
+# Issue#726
+interval_f(u, p) = u * u - 2.0 + p[1]
+uspan = (1.0, 2.0)
+prob_int = IntervalNonlinearProblem(interval_f, uspan)
+prob2 = @inferred IntervalNonlinearProblem remake(prob_int; p = [0])
+@test prob2.p == [2]