Skip to content

Fix output type of KernelRidge.predict#6354

Merged
rapids-bot[bot] merged 1 commit intorapidsai:branch-25.04from
jcrist:fix-kernel-ridge-output-type
Feb 23, 2025
Merged

Fix output type of KernelRidge.predict#6354
rapids-bot[bot] merged 1 commit intorapidsai:branch-25.04from
jcrist:fix-kernel-ridge-output-type

Conversation

@jcrist
Copy link
Copy Markdown
Member

@jcrist jcrist commented Feb 22, 2025

Previously this wasn't handled correctly, but was sometimes masked by GridSearchCV in the old test fixing things behind the scenes.

This fixes the bug, adds a test that doesn't require the accelerator, and changes the accelerator test to be both simpler (matching the smoketests in the rest of the file) and much faster (the GridSearchCV in the test made the test quite slow on my machine).

Fixup to the fix applied in #6327.

Previously this wasn't handled correctly, but was _sometimes_ masked by
`GridSearchCV` in the old test fixing things behind the scenes.

This fixes that, adds a test that doesn't require the accelerator, and
changes the accelerator test to be both simpler (matching the smoketests
in the rest of the file) and much faster (the `GridSearchCV` in the test
made the test quite slow on my machine).
@jcrist jcrist requested a review from a team as a code owner February 22, 2025 04:20
@jcrist jcrist requested review from divyegala and teju85 February 22, 2025 04:20
@github-actions github-actions Bot added the Cython / Python Cython or Python issue label Feb 22, 2025
@jcrist jcrist added bug Something isn't working non-breaking Non-breaking change cuml-cpu labels Feb 22, 2025
@dantegd
Copy link
Copy Markdown
Member

dantegd commented Feb 23, 2025

/merge

@rapids-bot rapids-bot Bot merged commit 6e05b00 into rapidsai:branch-25.04 Feb 23, 2025
dantegd added a commit to dantegd/cuml that referenced this pull request Feb 23, 2025
@jcrist jcrist deleted the fix-kernel-ridge-output-type branch February 23, 2025 02:15
dantegd added a commit to dantegd/cuml that referenced this pull request Feb 26, 2025
raydouglass pushed a commit that referenced this pull request Feb 28, 2025
PRs being backported: 

- [x] #6234
- [x] #6306
- [x] #6320
- [x] #6319
- [x] #6327
- [x] #6333
- [x] #6142 
- [x] #6223
- [x] #6235
- [x] #6317 
- [x] #6331
- [x] #6326
- [x] #6332
- [x] #6347
- [x] #6348
- [x] #6337
- [x] #6355
- [x] #6354
- [x] #6322
- [x] #6353
- [x] #6359
- [x] #6364
- [x] #6363
- [x] [FIL BATCH_TREE_REORG fix for SM90, 100 and
120](a3e419a)

---------

Co-authored-by: William Hicks <whicks@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working Cython / Python Cython or Python issue non-breaking Non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants