Skip to content

Commit 2710616

Browse files
authored
Add more valid options for compression= in to_parquet() (#815)
This matches the description in the API documentation for v2.1: https://pandas.pydata.org/pandas-docs/version/2.1/reference/api/pandas.DataFrame.to_parquet.html
1 parent 8f1ae07 commit 2710616

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pandas-stubs/core/frame.pyi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -386,7 +386,7 @@ class DataFrame(NDFrame, OpsMixin):
386386
self,
387387
path: FilePath | WriteBuffer[bytes],
388388
engine: ParquetEngine = ...,
389-
compression: Literal["snappy", "gzip", "brotli"] | None = ...,
389+
compression: Literal["snappy", "gzip", "brotli", "lz4", "zstd"] | None = ...,
390390
index: bool | None = ...,
391391
partition_cols: list[HashableT] | None = ...,
392392
storage_options: StorageOptions = ...,
@@ -397,7 +397,7 @@ class DataFrame(NDFrame, OpsMixin):
397397
self,
398398
path: None = ...,
399399
engine: ParquetEngine = ...,
400-
compression: Literal["snappy", "gzip", "brotli"] | None = ...,
400+
compression: Literal["snappy", "gzip", "brotli", "lz4", "zstd"] | None = ...,
401401
index: bool | None = ...,
402402
partition_cols: list[HashableT] | None = ...,
403403
storage_options: StorageOptions = ...,

0 commit comments

Comments
 (0)