From 96a09f3a81210549579b670254ed8862b87499df Mon Sep 17 00:00:00 2001
From: lxvm <lorenzo@vanmunoz.com>
Date: Mon, 26 Feb 2024 22:58:33 -0500
Subject: [PATCH] Revert "Update vector_of_array.jl"

This reverts commit 49d67427c5cfa94de80ad5231f6d930d4d4bc434.
---
 src/vector_of_array.jl | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/vector_of_array.jl b/src/vector_of_array.jl
index b8cc56b4..c563ce61 100644
--- a/src/vector_of_array.jl
+++ b/src/vector_of_array.jl
@@ -270,7 +270,13 @@ __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
-    return Array(A)
+    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)))
 end
 
 Base.@propagate_inbounds function _getindex(A::AbstractVectorOfArray{T, N},