Add correct dispatches for NonlinearLeastSquaresProblem

SciML · Oct 8, 2023 · b7dc21d · b7dc21d
1 parent 22c9bfc
commit b7dc21d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/solve.jl b/src/solve.jl
@@ -1098,6 +1098,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