Add correct dispatches for NonlinearLeastSquaresProblem

SciML · Sep 21, 2023 · 6f482ff · 6f482ff
1 parent 456cb3d
commit 6f482ff
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -76,7 +76,7 @@ PrecompileTools = "1"
 RecursiveArrayTools = "2"
 Reexport = "1.0"
 Requires = "1.0"
-SciMLBase = "1.93.3"
+SciMLBase = "2"
 SciMLOperators = "0.2, 0.3"
 Setfield = "0.8, 1"
 Static = "0.7, 0.8"

diff --git a/src/solve.jl b/src/solve.jl
@@ -1089,6 +1089,12 @@ function get_concrete_problem(prob::NonlinearProblem, isadapt; kwargs...)
     remake(prob; u0 = u0)
 end
 
+function get_concrete_problem(prob::NonlinearLeastSquaresProblem, isadapt; kwargs...)
+    u0 = get_concrete_u0(prob, isadapt, nothing, kwargs)
+    u0 = promote_u0(u0, prob.p, nothing)
+    remake(prob; u0 = u0)
+end
+
 function get_concrete_problem(prob::AbstractEnsembleProblem, isadapt; kwargs...)
     prob
 end