Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Python] Support group size and rows_per_file customization via write_dataset API #281

Merged
merged 2 commits into from
Nov 2, 2022

Conversation

eddyxu
Copy link
Contributor

@eddyxu eddyxu commented Nov 2, 2022

  • To bring more feature parity with pyarrow.dataset.write_dataset.
  • Fix docstring to make API docs look nicer.

@eddyxu eddyxu requested a review from changhiskhan November 2, 2022 15:54
@eddyxu eddyxu self-assigned this Nov 2, 2022
@eddyxu eddyxu added documentation Improvements or additions to documentation python labels Nov 2, 2022
Copy link
Contributor

@changhiskhan changhiskhan left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

one small nit

if mode == "create":
c_mode = CLanceDataset.WriteMode.CREATE
elif mode == "append":
if max_rows_per_file > 0:
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

let's add a simple check the max_rows_per_file >= max_rows_per_group?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

sgtm

@eddyxu eddyxu merged commit 5da61d0 into main Nov 2, 2022
@eddyxu eddyxu deleted the lei/fix_pydocs branch November 2, 2022 16:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
documentation Improvements or additions to documentation python
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants