Skip to content

Don't exclude sklearn tests from acceleration#6565

Merged
rapids-bot[bot] merged 1 commit intorapidsai:branch-25.06from
jcrist:accelerator-dont-exclude-sklearn-tests
Apr 22, 2025
Merged

Don't exclude sklearn tests from acceleration#6565
rapids-bot[bot] merged 1 commit intorapidsai:branch-25.06from
jcrist:accelerator-dont-exclude-sklearn-tests

Conversation

@jcrist
Copy link
Copy Markdown
Member

@jcrist jcrist commented Apr 21, 2025

When we rewrote cuml.accel's mechanism to use a finder and loader (and include a denylist for excluding certain modules from acceleration), we accidentally excluded the sklearn tests from acceleration. We don't want to do this since we run the sklearn test suite with cuml.accel enabled to validate our implementation.

This PR:

  • Renames denylist to exclude
  • Makes exclude optionally take a callable instead of a sequence
  • Updates our exclude logic for cuml.accel to exempt modules matching sklearn.*.tests.* from exclusion.

When we rewrote `cuml.accel`'s mechanism to use a finder and loader (and
include a `denylist` for excluding certain modules from acceleration),
we accidentally excluded the `sklearn` tests from acceleration. We don't
want to do this since we run the sklearn test suite with `cuml.accel`
enabled to validate our implementation.

This PR:
- Renames `denylist` to `exclude`
- Makes `exclude` optionally take a callable instead of a sequence
- Updates our exclude logic for `cuml.accel` to exempt modules matching
  `sklearn.*.tests.*` from exclusion.
@jcrist jcrist requested a review from a team as a code owner April 21, 2025 22:20
@jcrist jcrist requested review from csadorf and teju85 April 21, 2025 22:20
@github-actions github-actions Bot added the Cython / Python Cython or Python issue label Apr 21, 2025
@jcrist jcrist added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change cuml-accel Issues related to cuml.accel labels Apr 21, 2025
Copy link
Copy Markdown
Contributor

@csadorf csadorf left a comment

Choose a reason for hiding this comment

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

Awesome! Thanks for addressing this so quickly!

@jcrist
Copy link
Copy Markdown
Member Author

jcrist commented Apr 22, 2025

/merge

@rapids-bot rapids-bot Bot merged commit 39cabba into rapidsai:branch-25.06 Apr 22, 2025
75 of 76 checks passed
@jcrist jcrist self-assigned this Apr 22, 2025
Ofek-Haim pushed a commit to Ofek-Haim/cuml that referenced this pull request May 13, 2025
When we rewrote `cuml.accel`'s mechanism to use a finder and loader (and include a `denylist` for excluding certain modules from acceleration), we accidentally excluded the `sklearn` tests from acceleration. We don't want to do this since we run the sklearn test suite with `cuml.accel` enabled to validate our implementation.

This PR:
- Renames `denylist` to `exclude`
- Makes `exclude` optionally take a callable instead of a sequence
- Updates our exclude logic for `cuml.accel` to exempt modules matching `sklearn.*.tests.*` from exclusion.

Authors:
  - Jim Crist-Harif (https://github.com/jcrist)

Approvers:
  - Simon Adorf (https://github.com/csadorf)

URL: rapidsai#6565
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cuml-accel Issues related to cuml.accel Cython / Python Cython or Python issue improvement Improvement / enhancement to an existing function non-breaking Non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants