[
https://issues.apache.org/jira/browse/ARROW-15484?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=17483746#comment-17483746
]
Martin Thøgersen edited comment on ARROW-15484 at 1/28/22, 12:37 PM:
---------------------------------------------------------------------
Thanks [~westonpace] , that solved my issue.
I think the reference to {{write_table}} in the docstring confused me, because
it's quite overloaded in various sub-modules.
Working code:
{code:python}
import pyarrow.dataset as ds
ds.write_dataset(data=table,
base_dir='foo',
basename_template="prefix-{i}.parquet",
format="parquet",
existing_data_behavior="error"
)
{code}
was (Author: JIRAUSER284229):
Thanks [~westonpace] , that solved my issue.
I think the reference to {{write_table}} in the docstring confused me, because
it's quite overloaded in various sub-modules.
> [Python] kwargs fails for pyarrow.parquet.write_to_dataset()
> ------------------------------------------------------------
>
> Key: ARROW-15484
> URL: https://issues.apache.org/jira/browse/ARROW-15484
> Project: Apache Arrow
> Issue Type: Bug
> Components: Parquet, Python
> Affects Versions: 6.0.1
> Reporter: Martin Thøgersen
> Priority: Major
>
> When supplying `kwargs` such as `basename_template` or
> `existing_data_behavior` to `pyarrow.parquet.write_to_dataset()`, it fails as
> below.
>
> {code:python}
> import pandas as pd
> import pyarrow as pa
> import pyarrow.parquet as pq
> import pyarrow.dataset as ds
> df = pd.DataFrame({
> 'int': [1, 2],
> 'str': ['a', 'b']
> })
> table = pa.Table.from_pandas(df)
> """
> **kwargs : dict,
> Additional kwargs for write_table function. See docstring for write_table
> or ParquetWriter for more information.
> """
> pq.write_to_dataset(table, root_path='foo',
> use_legacy_dataset=False,
> # kwargs:
> basename_template="prefix-{i}.parquet",
> existing_data_behavior="error"
> )
> {code}
> {noformat}
> TypeError Traceback (most recent call last)
> ...test.py in <module>
> 16 Additional kwargs for write_table function. See docstring for
> write_table or ParquetWriter for more information.
> 17 """
> ---> 18 pq.write_to_dataset(table, root_path='foo',
> 19 use_legacy_dataset=False,
> 20 # kwargs:
> ...lib/python3.8/site-packages/pyarrow/parquet.py in write_to_dataset(table,
> root_path, partition_cols, partition_filename_cb, filesystem,
> use_legacy_dataset, **kwargs)
> 2144 # map format arguments
> 2145 parquet_format = ds.ParquetFileFormat()
> -> 2146 write_options = parquet_format.make_write_options(**kwargs)
> 2147
> 2148 # map old filesystems to new one
> ...lib/python3.8/site-packages/pyarrow/_dataset.pyx in
> pyarrow._dataset.ParquetFileFormat.make_write_options()
> ...lib/python3.8/site-packages/pyarrow/_dataset.pyx in
> pyarrow._dataset.ParquetFileWriteOptions.update()
> TypeError: unexpected parquet write option: basename_template
> {noformat}
--
This message was sent by Atlassian Jira
(v8.20.1#820001)
