Override iterate for BatchView

JuliaML · Feb 22, 2022 · c0a19e3 · c0a19e3
1 parent a771cea
commit c0a19e3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/batchview.jl b/src/batchview.jl
@@ -119,6 +119,10 @@ function Base.getindex(A::BatchView, is::AbstractVector)
     obsview(A.data, obsindices)
 end
 
+# override AbstractDataContainer default
+Base.iterate(A::BatchView, state = 1) =
+    (state > numobs(A)) ? nothing : (A[state], state + 1)
+
 obsview(A::BatchView) = A
 obsview(A::BatchView, i) = A[i]