allow_empty is the default in >=zstandard-0.9 (#355)

* allow_empty is the default in >=zstandard-0.9 * add comment
dask · Aug 6, 2018 · 8182b7b · 8182b7b
1 parent 14a6af6
commit 8182b7b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/fastparquet/compression.py b/fastparquet/compression.py
@@ -92,7 +92,11 @@ def zstd_decompress(data, uncompressed_size):
         def zstd_compress(data, **kwargs):
             kwargs['write_content_size'] = False
             cctx = zstd.ZstdCompressor(**kwargs)
-            return cctx.compress(data, allow_empty=True)
+            try:
+                return cctx.compress(data, allow_empty=True)
+            except TypeError:
+                # zstandard-0.9 removed allow_empy and made it the default.
+                return cctx.compress(data)
         def zstd_decompress(data, uncompressed_size):
             dctx = zstd.ZstdDecompressor()
             return dctx.decompress(data, max_output_size=uncompressed_size)