From 3ec080b95ca5095fb68b5c31ee0e5e8387eb8b01 Mon Sep 17 00:00:00 2001 From: Judah Rand <17158624+judahrand@users.noreply.github.com> Date: Wed, 30 Aug 2023 16:49:10 +0100 Subject: [PATCH] Add remaining missing args --- python/pyarrow/_dataset_parquet.pyx | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/pyarrow/_dataset_parquet.pyx b/python/pyarrow/_dataset_parquet.pyx index 2177cb8c9f32f..cf5c44c1c964a 100644 --- a/python/pyarrow/_dataset_parquet.pyx +++ b/python/pyarrow/_dataset_parquet.pyx @@ -596,6 +596,8 @@ cdef class ParquetFileWriteOptions(FileWriteOptions): column_encoding=self._properties["column_encoding"], data_page_version=self._properties["data_page_version"], encryption_properties=self._properties["encryption_properties"], + write_batch_size=self._properties["write_batch_size"], + dictionary_pagesize_limit=self._properties["dictionary_pagesize_limit"], write_page_index=self._properties["write_page_index"], ) @@ -634,6 +636,8 @@ cdef class ParquetFileWriteOptions(FileWriteOptions): allow_truncated_timestamps=False, use_compliant_nested_type=True, encryption_properties=None, + write_batch_size=None, + dictionary_pagesize_limit=None, write_page_index=False, ) self._set_properties()