Merge pull request #33 from avik-pal/patch-1

Fix SSrootfind
SciML · Sep 24, 2021 · 0036db6 · 0036db6
2 parents 58387c0 + 07fef70
commit 0036db6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/solve.jl b/src/solve.jl
@@ -41,7 +41,7 @@ function DiffEqBase.__solve(prob::DiffEqBase.AbstractSteadyStateProblem,
   # f = (u) -> (f!(du,u); du) # out-of-place version
 
   if typeof(alg) <: SSRootfind
-    u = alg.nlsolve(f!,u0,abstol)
+    u = reshape(alg.nlsolve(f!,u0,abstol), sizeu)
     resid = similar(u)
     f!(resid,u)
     DiffEqBase.build_solution(prob,alg,u,resid;retcode = :Success)