Shortcut __restructure_sol

SciML · Dec 16, 2024 · 87a1e10 · 87a1e10
1 parent fcb3ee1
commit 87a1e10
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/BoundaryValueDiffEqCore/src/utils.jl b/lib/BoundaryValueDiffEqCore/src/utils.jl
@@ -240,9 +240,11 @@ __vec_bc(sol, p, bc, u_size) = vec(bc(reshape(sol, u_size), p))
 
 # Restructure Solution
 function __restructure_sol(sol::AbstractVectorOfArray, u_size)
+    (size(first(sol)) == u_size) && return sol
     return VectorOfArray(map(Base.Fix2(reshape, u_size), sol))
 end
 function __restructure_sol(sol::AbstractArray{<:AbstractArray}, u_size)
+    (size(first(sol)) == u_size) && return sol
     return map(Base.Fix2(reshape, u_size), sol)
 end