Skip to content

Commit b1db46c

Browse files
committed
fix: clarify compression level documentation for ZSTD in write_parquet method
1 parent 0c3fed9 commit b1db46c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/datafusion/dataframe.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -632,9 +632,10 @@ def write_parquet(
632632
recommended range is 1 to 22, with the default being 3. Higher levels
633633
provide better compression but slower speed.
634634
"""
635-
# default compression level to 3 for ZSTD
636635
if compression == "ZSTD":
637636
if compression_level is None:
637+
# Default compression level for ZSTD is 3 as per
638+
# https://facebook.github.io/zstd/zstd_manual.html
638639
compression_level = 3
639640
elif not (1 <= compression_level <= 22):
640641
raise ValueError("Compression level for ZSTD must be between 1 and 22")

0 commit comments

Comments
 (0)