Preserve dtype for NumPy array-protocol inputs - #8432
Conversation
|
No actionable comments were generated in the recent review. 🎉 ℹ️ Recent review info⚙️ Run configurationConfiguration used: Path: .coderabbit.yaml Review profile: CHILL Plan: Enterprise Run ID: 📒 Files selected for processing (3)
💤 Files with no reviewable changes (1)
🚧 Files skipped from review as they are similar to previous changes (2)
📝 WalkthroughSummary by CodeRabbit
Walkthrough
ChangesArray validation and compatibility
Estimated code review effort: 2 (Simple) | ~10 minutes Possibly related PRs
Suggested reviewers: 🚥 Pre-merge checks | ✅ 4 | ❌ 1❌ Failed checks (1 warning)
✅ Passed checks (4 passed)
✨ Finishing Touches🧪 Generate unit tests (beta)
Comment |
betatim
left a comment
There was a problem hiding this comment.
This looks like a reasonable fix.
jcrist
left a comment
There was a problem hiding this comment.
Fix looks good to me! A bit disappointing that this didn't fix the issues in test_sklearn_compatibility.py though.
|
/merge |
Closes #8430.
Problem : The issue seems to be unrelated to the GB300 platform. Scikit-learn’s array-like compatibility checks wrap float64 data in an object without a direct dtype attribute. cuML consequently defaulted the wrapped input to float32, producing different results from the original float64 input.
Solution : Normalize objects implementing array before dtype selection to preserve their represented dtype, and remove the stale xfails.