Update src/solutions/nonlinear_solutions.jl

Co-authored-by: Yingbo Ma <[email protected]>
SciML · Feb 3, 2021 · c37ff96 · c37ff96
1 parent efef10e
commit c37ff96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/solutions/nonlinear_solutions.jl b/src/solutions/nonlinear_solutions.jl
@@ -23,7 +23,7 @@ function build_solution(prob::AbstractNonlinearProblem,
                         kwargs...)
 
   T = eltype(eltype(u))
-  N = length((size(u)...,))
+  N = ndims(u)
 
   NonlinearSolution{T,N,typeof(u),typeof(resid),
                     typeof(prob),typeof(alg),typeof(original),typeof(left)}(