Merge pull request #1288 from JuliaData/jq/datastreams

Update DataStreams allocate for CategoricalString
JuliaData · Nov 26, 2017 · f031a3d · f031a3d
2 parents 2574552 + b5d6c61
commit f031a3d
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{CategoricalString{R}}, rows, ref) where {R} = CategoricalArray{String, 1, R}(rows)
+allocate(::Type{Union{CategoricalString{R}, Missing}}, rows, ref) where {R} = CategoricalArray{Union{String, Missing}, 1, R}(rows)
 allocate(::Type{CategoricalValue{T, R}}, rows, ref) where {T, R} =
     CategoricalArray{T, 1, R}(rows)
 allocate(::Type{Union{Missing, CategoricalValue{T, R}}}, rows, ref) where {T, R} =