Add an allocate method for CategoricalArrays for DataStreams

JuliaData · Sep 14, 2017 · 5488484 · 5488484
1 parent fb72599
commit 5488484
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/abstractdataframe/io.jl b/src/abstractdataframe/io.jl
@@ -230,6 +230,8 @@ Data.streamtypes(::Type{DataFrame}) = [Data.Column, Data.Field]
 Data.weakrefstrings(::Type{DataFrame}) = true
 
 allocate(::Type{T}, rows, ref) where {T} = Vector{T}(rows)
+allocate(::Type{T}, rows, ref) where {T <: Union{CategoricalValue, Null}} =
+    CategoricalArray{CategoricalArrays.unwrap_catvalue_type(T)}(rows)
 allocate(::Type{T}, rows, ref) where {T <: Union{WeakRefString, Null}} =
     WeakRefStringArray(ref, T, rows)