Shim IteratorSize for DataValueRowIterator (#44)

JuliaData · Nov 13, 2018 · 61dae02 · 61dae02
1 parent c3a2062
commit 61dae02
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/datavalues.jl b/src/datavalues.jl
@@ -29,8 +29,11 @@ function datavaluerows(x)
     return DataValueRowIterator(datavaluetype(Tables.schema(r)), r)
 end
 
+_iteratorsize(x) = x
+_iteratorsize(::Base.HasShape{1}) = Base.HasLength()
+
 Base.eltype(rows::DataValueRowIterator{NT, S}) where {NT, S} = NT
-Base.IteratorSize(::Type{DataValueRowIterator{NT, S}}) where {NT, S} = Base.IteratorSize(S)
+Base.IteratorSize(::Type{DataValueRowIterator{NT, S}}) where {NT, S} = _iteratorsize(Base.IteratorSize(S))
 Base.length(rows::DataValueRowIterator) = length(rows.x)
 
 function Base.iterate(rows::DataValueRowIterator{NT, S}, st=()) where {NT <: NamedTuple{names}, S} where {names}