use length for chunk iterator state rather than .taken (#81)

meggart · Dec 15, 2022 · 37dfacd · 37dfacd
1 parent 68c8150
commit 37dfacd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/iterator.jl b/src/iterator.jl
@@ -50,14 +50,14 @@ macro implement_iteration(t)
         function Base.iterate(a::$t, i)
             datacur, bi, bstate = i
             (chunkiter, innerinds) = bstate
-            cistateold = chunkiter.taken
+            cistateold = length(chunkiter)
             biter = iterate(bi, bstate)
             if biter === nothing
                 return nothing
             end
             innernow, bstatenew = biter
             (chunkiter, innerinds) = bstatenew
-            if chunkiter.taken !== cistateold
+            if length(chunkiter) !== cistateold
                 curchunk = innerinds.itr.indices
                 datacur = OffsetArray(a[curchunk...], innerinds.itr)
             end