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

Expand JIT groupby test suite #13813

Merged

Conversation

brandon-b-miller
Copy link
Contributor

@brandon-b-miller brandon-b-miller commented Aug 3, 2023

This PR reorganizes and expands the test suite for groupby apply functions using the JIT engine to include nan cases and cases where the groups are larger than a single thread block.

@brandon-b-miller brandon-b-miller added feature request New feature or request Python Affects Python cuDF API. non-breaking Non-breaking change labels Aug 3, 2023
@brandon-b-miller brandon-b-miller self-assigned this Aug 3, 2023
@ajschmidt8 ajschmidt8 mentioned this pull request Aug 3, 2023
3 tasks
@brandon-b-miller brandon-b-miller marked this pull request as ready for review August 14, 2023 16:42
@brandon-b-miller brandon-b-miller requested a review from a team as a code owner August 14, 2023 16:42
@brandon-b-miller
Copy link
Contributor Author

Marking this ready for review - I am going to work on solving some of the xfails that came up when putting this PR, so hopefully some of them won't even be necessary by merge time.

Copy link
Contributor

@bdice bdice left a comment

Choose a reason for hiding this comment

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

@brandon-b-miller Just to clarify, you're planning to merge this to introduce new xfailed tests, then fix them in subsequent PRs?

python/cudf/cudf/tests/test_groupby.py Outdated Show resolved Hide resolved
python/cudf/cudf/core/groupby/groupby.py Outdated Show resolved Hide resolved
@brandon-b-miller
Copy link
Contributor Author

@brandon-b-miller Just to clarify, you're planning to merge this to introduce new xfailed tests, then fix them in subsequent PRs?

Correct, this is my plan, based on not knowing how long it might take to fix some of these.

Co-authored-by: Bradley Dice <bdice@bradleydice.com>
python/cudf/cudf/tests/test_groupby.py Outdated Show resolved Hide resolved
python/cudf/cudf/tests/test_groupby.py Outdated Show resolved Hide resolved
python/cudf/cudf/tests/test_groupby.py Outdated Show resolved Hide resolved
python/cudf/cudf/tests/test_groupby.py Outdated Show resolved Hide resolved
python/cudf/cudf/tests/test_groupby.py Outdated Show resolved Hide resolved
@brandon-b-miller brandon-b-miller changed the base branch from branch-23.10 to branch-24.02 December 15, 2023 16:58
Copy link

copy-pr-bot bot commented Dec 15, 2023

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@brandon-b-miller
Copy link
Contributor Author

/ok to test

@brandon-b-miller
Copy link
Contributor Author

/ok to test

Copy link
Contributor

@bdice bdice left a comment

Choose a reason for hiding this comment

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

A couple small cleanup requests, otherwise LGTM!

np.random.seed(0)
def groupby_jit_data_small():
"""
Small dataset for testing groupby apply with jit
Copy link
Contributor

Choose a reason for hiding this comment

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

Can we write this comment with complete sentences / capitalization? It also feels like it's wrapped pretty harshly, like 50-60 chars / line? Same request for other comments below.

python/cudf/cudf/tests/test_groupby.py Outdated Show resolved Hide resolved
@brandon-b-miller
Copy link
Contributor Author

/ok to test

@brandon-b-miller
Copy link
Contributor Author

/ok to test

@brandon-b-miller
Copy link
Contributor Author

/merge

@rapids-bot rapids-bot bot merged commit 8df33ee into rapidsai:branch-24.02 Jan 10, 2024
67 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request New feature or request non-breaking Non-breaking change Python Affects Python cuDF API.
Projects
Archived in project
Development

Successfully merging this pull request may close these issues.

None yet

3 participants