Merge pull request #96 from JuliaIO/jq/tables

Accept any Tables.jl-compatible source to write_parquet
JuliaIO · Jul 3, 2020 · a6a3741 · a6a3741
2 parents e0618d4 + ce3eb58
commit a6a3741
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/writer.jl b/src/writer.jl
@@ -460,9 +460,8 @@ tbl                 -   A Tables.jl columnaccessible table e.g. a DataFrame
 compression_code    -   Default "SNAPPY". The compression codec. The supported
                         values are "UNCOMPRESSED", "SNAPPY", "ZSTD", "GZIP"
 """
-function write_parquet(path, tbl; compression_codec = "SNAPPY")
-    # tbl needs to be iterable by column as parquet is a columnar format
-    @assert Tables.columnaccess(tbl)
+function write_parquet(path, x; compression_codec = "SNAPPY")
+    tbl = Tables.columns(x)
 
     # check that all types are supported
     sch = Tables.schema(tbl)