Skip to content

Conversation

@vyasr
Copy link
Contributor

@vyasr vyasr commented Sep 2, 2025

This PR adds deprecation warnings to pynvml and pynvml_utils indicating that these packages are going away. For pynvml_utils we could use a simpler approach of just throwing the warning inside the package's __init__.py file, but for pynvml we have to use the pth approach since it's just a metapackage relying on nvidia-ml-py for all its code, so we may as well reuse the finder's find_spec method for both warnings.

Copy link

@pentschev pentschev left a comment

Choose a reason for hiding this comment

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

I think this looks fine overall, to be completely honest I'm not familiar enough with pth to judge what's being added in setup.py, but I trust you have done what you thought was best and tested as much as you could to ensure it works, therefore I'll approve this. Thanks @vyasr !

@vyasr
Copy link
Contributor Author

vyasr commented Sep 4, 2025

Thanks Peter! I tested in a few environments and it seemed good. I'll defer to @rjzamora on reviewing/approving/merging when ready. We'll need to make a new release once it's merged.

@rjzamora rjzamora merged commit f0bd53f into gpuopenanalytics:master Sep 4, 2025
@vyasr vyasr deleted the chore/deprecation branch September 10, 2025 19:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants