Update vector_of_array.jl

Fix SciML#354
lxvm · Feb 19, 2024 · 49d6742 · 49d6742
1 parent 87337d8
commit 49d6742
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/src/vector_of_array.jl b/src/vector_of_array.jl
@@ -270,13 +270,7 @@ __parameterless_type(T) = Base.typename(T).wrapper
 Base.@propagate_inbounds function _getindex(A::AbstractVectorOfArray{T, N},
     ::NotSymbolic, I::Colon...) where {T, N}
     @assert length(I) == ndims(A.u[1]) + 1
-    vecs = if N == 1
-        A.u
-    else
-        vec.(A.u)
-    end
-    return Adapt.adapt(__parameterless_type(T),
-        reshape(reduce(hcat, vecs), size(A.u[1])..., length(A.u)))
+    return Array(A)
 end
 
 Base.@propagate_inbounds function _getindex(A::AbstractVectorOfArray{T, N},