From 1a4f35ce41429a48e4c1c7255d73c3eb764cc245 Mon Sep 17 00:00:00 2001
From: Peter Andreas Entschev <peter@entschev.com>
Date: Thu, 14 Nov 2024 23:56:00 -0800
Subject: [PATCH] Set PyNVML upper pin

---
 dependencies.yaml | 2 +-
 pyproject.toml    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/dependencies.yaml b/dependencies.yaml
index 86ef4c25..4cb6e261 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -137,7 +137,7 @@ dependencies:
       - output_types: [conda, requirements, pyproject]
         packages:
           - numpy>=1.23,<3.0a0
-          - pynvml>=11.4.1
+          - pynvml>=11.4.1,<12.0.0a0
   depends_on_cupy:
     common:
       - output_types: conda
diff --git a/pyproject.toml b/pyproject.toml
index 1c7eb1af..86b22e4e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ license = { text = "BSD-3-Clause" }
 requires-python = ">=3.10"
 dependencies = [
     "numpy>=1.23,<3.0a0",
-    "pynvml>=11.4.1",
+    "pynvml>=11.4.1,<12.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
 classifiers = [
     "Intended Audience :: Developers",