Skip to content

Pin sklearn to < 1.8.0 for cuml 25.10 - #1368

Closed
praateekmahajan wants to merge 20 commits into
NVIDIA-NeMo:r1.1.0from
praateekmahajan:praateek/sklearn-version
Closed

Pin sklearn to < 1.8.0 for cuml 25.10#1368
praateekmahajan wants to merge 20 commits into
NVIDIA-NeMo:r1.1.0from
praateekmahajan:praateek/sklearn-version

Conversation

@praateekmahajan

Copy link
Copy Markdown
Contributor

Description

As title, since cuml 25.10 is incompatible and the import fails if user installs sklearn 1.8.0+ unintentionally.

Usage

# Add snippet demonstrating usage

Checklist

  • I am familiar with the Contributing Guide.
  • New or Existing tests cover these changes.
  • The documentation is up to date with these changes.

suiyoubi and others added 5 commits January 12, 2026 14:31
Signed-off-by: Ao Tang <aot@nvidia.com>
Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
…orStage (#1350)

Signed-off-by: Abhinav Garg <abhinavg@stanford.edu>
Co-authored-by: Sarah Yurick <53962159+sarahyurick@users.noreply.github.com>
Co-authored-by: Ayush Dattagupta <ayushdg95@gmail.com>
Signed-off-by: Praateek <praateekm@gmail.com>

@greptile-apps greptile-apps Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

1 file reviewed, 1 comment

Edit Code Review Agent Settings | Greptile

Comment thread pyproject.toml
"pylibraft-cu12==25.10.*",
"raft-dask-cu12==25.10.*",
"rapidsmpf-cu12==25.10.*",
"scikit-learn<1.8.0", # cuml 25.10 is incompatible with sklearn 1.8.0+

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

syntax: Duplicate scikit-learn<1.8.0 constraint - already exists on line 79

Suggested change
"scikit-learn<1.8.0", # cuml 25.10 is incompatible with sklearn 1.8.0+

@ayushdg
ayushdg changed the base branch from main to r1.1.0 January 20, 2026 21:16
@praateekmahajan

Copy link
Copy Markdown
Contributor Author

Closing in favor of #1405

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

r1.1.0 Pick this label for auto cherry-picking into r1.1.0

Projects

None yet

Development

Successfully merging this pull request may close these issues.

8 participants