-
Notifications
You must be signed in to change notification settings - Fork 75
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
Remove NumPy <2 pin #297
Remove NumPy <2 pin #297
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks Sebastian! 🙏
Had a suggestion below
Pushed a small change to line up the dependencies across conda and wheels. Hope that is ok As this depends on RAFT, CI may fail until we get PR ( rapidsai/raft#2414 ) through. We can restart after that is done |
Some jobs were failing with what looked like temporary network errors, so I restarted CI here. |
Restarting CI now that RAFT has been rebuilt with NumPy 2 support: rapidsai/raft#2414 |
Nice, seems like this is passing now and using NumPy 2.0.1 (I also briefly scanned the code to see if the most likely NumPy 2.1 change can't affect it -- related to |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good, I double-checked that there's no need for a build dependency on numpy
here, that the cuda-python
dependency added is necessary, and that we're seeing numpy
2.x in tests.
/merge |
Thanks Sebastian and James! 🙏 |
This PR removes the NumPy<2 pin which is expected to work for
RAPIDS projects once CuPy 13.3.0 is released (CuPy 13.2.0 had
some issues preventing the use with NumPy 2).