make A[:, end] error for ragged AbstractVectorOfArray when the length…

…s are not equal
SciML · Jul 12, 2023 · 6df1fc7 · 6df1fc7
1 parent 7a3c3e4
commit 6df1fc7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/vector_of_array.jl b/src/vector_of_array.jl
@@ -541,3 +541,10 @@ end
 end
 unpack_args_voa(i, args::Tuple{Any}) = (unpack_voa(args[1], i),)
 unpack_args_voa(::Any, args::Tuple{}) = ()
+
+function Base.lastindex(A::AbstractVectorOfArray, i=1)
+    i == 1 && return length(A)
+    len1 = lastindex(A[1], i-1)
+    all(x->isequal(lastindex(x, i-1), len1), A) && return len1
+    throw(ArgumentError("Can not get lastindex of an AbstractVectorOfArray for dimensions other than 1"))
+end