fix: return type inference

Author: avik-pal

lib/SimpleNonlinearSolve/src/lbroyden.jl

@@ -62,10 +62,11 @@ end
 
     # For scalar problems / if the threshold is larger than problem size just use Broyden
     if x isa Number || length(x) ≤ η
-        return SciMLBase.__solve(
+        sol = SciMLBase.__solve(
             prob, SimpleBroyden(; alg.linesearch), args...;
             abstol, reltol, maxiters, termination_condition, kwargs...
         )
+        return Utils.nonlinear_solution_new_alg(sol, alg)
     end
 
     fx = NLBUtils.evaluate_f(prob, x)

lib/SimpleNonlinearSolve/src/utils.jl

@@ -174,4 +174,13 @@ function compute_hvvp(prob, autodiff, fx, x, dir)
     return only(DI.pushforward(jvp_fn, autodiff, x, (dir,), Constant(prob.p)))
 end
 
+function nonlinear_solution_new_alg(
+        sol::SciMLBase.NonlinearSolution{T, N, uType, R, P, A, O, uType2, S, Tr}, alg
+) where {T, N, uType, R, P, A, O, uType2, S, Tr}
+    return SciMLBase.NonlinearSolution{T, N, uType, R, P, typeof(alg), O, uType2, S, Tr}(
+        sol.u, sol.resid, sol.prob, alg, sol.retcode, sol.original, sol.left, sol.right,
+        sol.stats, sol.trace
+    )
+end
+
 end