Fix for bytes/str discrepancy after PyNVML update

dask · Feb 15, 2023 · fca95c0 · fca95c0
1 parent ca10a7c
commit fca95c0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/distributed/diagnostics/nvml.py b/distributed/diagnostics/nvml.py
@@ -317,6 +317,8 @@ def _get_memory_total(h):
 def _get_name(h):
     try:
         return pynvml.nvmlDeviceGetName(h).decode()
+    except AttributeError:
+        return pynvml.nvmlDeviceGetName(h)
     except pynvml.NVMLError_NotSupported:
         return None