From 5238822ba1ee6d8c5b1df3047accc5631a736173 Mon Sep 17 00:00:00 2001
From: Peter Andreas Entschev <peter@entschev.com>
Date: Thu, 21 Nov 2024 02:47:26 +0100
Subject: [PATCH] Set PyNVML upper pin (#323)

The plan is for `pynvml` to become a metapackage in version 12, where it not anymore ship NVML bindings but instead install `nvidia-ml-py` to serve the official NVML bindings. To provide a better future support, setting an upper pin for `pynvml` is a safer choice.

Authors:
  - Peter Andreas Entschev (https://github.com/pentschev)

Approvers:
  - https://github.com/jakirkham

URL: https://github.com/rapidsai/ucxx/pull/323
---
 conda/environments/all_cuda-118_arch-x86_64.yaml | 2 +-
 conda/environments/all_cuda-125_arch-x86_64.yaml | 2 +-
 conda/recipes/ucxx/meta.yaml                     | 2 +-
 dependencies.yaml                                | 2 +-
 python/ucxx/pyproject.toml                       | 2 +-
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
index d154ac81..f1ac9615 100644
--- a/conda/environments/all_cuda-118_arch-x86_64.yaml
+++ b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -30,7 +30,7 @@ dependencies:
 - pip
 - pkg-config
 - pre-commit
-- pynvml>=11.4.1
+- pynvml>=11.4.1,<12.0.0a0
 - pytest-asyncio
 - pytest-rerunfailures
 - pytest==7.*
diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml
index 2aa1d168..c87c5c1e 100644
--- a/conda/environments/all_cuda-125_arch-x86_64.yaml
+++ b/conda/environments/all_cuda-125_arch-x86_64.yaml
@@ -30,7 +30,7 @@ dependencies:
 - pip
 - pkg-config
 - pre-commit
-- pynvml>=11.4.1
+- pynvml>=11.4.1,<12.0.0a0
 - pytest-asyncio
 - pytest-rerunfailures
 - pytest==7.*
diff --git a/conda/recipes/ucxx/meta.yaml b/conda/recipes/ucxx/meta.yaml
index 8d9e3f13..3b391983 100644
--- a/conda/recipes/ucxx/meta.yaml
+++ b/conda/recipes/ucxx/meta.yaml
@@ -240,7 +240,7 @@ outputs:
         - {{ pin_subpackage('libucxx', exact=True) }}
         - {{ pin_compatible('rmm', max_pin='x.x') }}
         - numpy>=1.23,<3.0a0
-        - pynvml >=11.4.1
+        - pynvml >=11.4.1,<12.0.0a0
       run_constrained:
         - cupy >=9.5.0
         - numba >=0.57.0
diff --git a/dependencies.yaml b/dependencies.yaml
index fc7d0703..1d83d929 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -283,7 +283,7 @@ dependencies:
       - output_types: [conda, requirements, pyproject]
         packages:
           - &numpy numpy>=1.23,<3.0a0
-          - pynvml>=11.4.1
+          - pynvml>=11.4.1,<12.0.0a0
   run_python_distributed_ucxx:
     common:
       - output_types: [conda, requirements, pyproject]
diff --git a/python/ucxx/pyproject.toml b/python/ucxx/pyproject.toml
index 730fc0cb..0cf5feab 100644
--- a/python/ucxx/pyproject.toml
+++ b/python/ucxx/pyproject.toml
@@ -21,7 +21,7 @@ requires-python = ">=3.10"
 dependencies = [
     "libucxx==0.41.*,>=0.0.0a0",
     "numpy>=1.23,<3.0a0",
-    "pynvml>=11.4.1",
+    "pynvml>=11.4.1,<12.0.0a0",
     "rmm==24.12.*,>=0.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 classifiers = [