From d281b83d5a5420e55b6679c0b96d8ac87b678e71 Mon Sep 17 00:00:00 2001
From: Lawrence Mitchell <lmitchell@nvidia.com>
Date: Wed, 22 Feb 2023 18:06:39 +0000
Subject: [PATCH] Pin to pynvml < 11.5 (#1123)

PyNVML 11.5 makes a API-breaking change to the return types of (at least) nvmlDeviceGetName (producing str rather than bytes). On the development branch this was fixed by #1118. To fix in 23.02, instead pin to the previously working version, rather than backporting.

Authors:
   - Lawrence Mitchell (https://github.com/wence-)

Approvers:
   - Ray Douglass (https://github.com/raydouglass)
---
 dependencies.yaml | 2 +-
 pyproject.toml    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/dependencies.yaml b/dependencies.yaml
index 9b471e6a..f2fc9365 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -100,7 +100,7 @@ dependencies:
           - numba>=0.54
           - numpy>=1.18.0
           - pandas>=1.0
-          - pynvml>=11.0.0
+          - pynvml>=11.0.0,<11.5
           - zict>=0.1.3
   test_python:
     common:
diff --git a/pyproject.toml b/pyproject.toml
index 58f156bb..5dafea82 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -21,7 +21,7 @@ requires-python = ">=3.8"
 dependencies = [
     "dask ==2023.1.1",
     "distributed ==2023.1.1",
-    "pynvml >=11.0.0",
+    "pynvml >=11.0.0,<11.5",
     "numpy >=1.18.0",
     "numba >=0.54",
     "pandas >=1.0",