From 4833f201ad2eb4bc07e5d561bd17ac36f4ecd546 Mon Sep 17 00:00:00 2001
From: Xuehai Pan <XuehaiPan@pku.edu.cn>
Date: Sat, 1 Jun 2024 11:24:47 +0000
Subject: [PATCH] deps(nvidia-ml-py): add `nvidia-ml-py` 12.555.43 to support
 list

---
 nvitop/version.py | 1 +
 pyproject.toml    | 2 +-
 requirements.txt  | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/nvitop/version.py b/nvitop/version.py
index 7e000f5..4192319 100644
--- a/nvitop/version.py
+++ b/nvitop/version.py
@@ -75,6 +75,7 @@ PYNVML_VERSION_CANDIDATES = (
     '12.535.133',
     '12.535.161',
     '12.550.52',
+    '12.555.43',
 )
 """The list of supported ``nvidia-ml-py`` versions.
 See also: `nvidia-ml-py's Release History <https://pypi.org/project/nvidia-ml-py/#history>`_.
diff --git a/pyproject.toml b/pyproject.toml
index a25b186..1f8d5df 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -46,7 +46,7 @@ classifiers = [
 ]
 dependencies = [
     # Sync with nvitop/version.py and requirements.txt
-    "nvidia-ml-py >= 11.450.51, < 12.551.0a0",
+    "nvidia-ml-py >= 11.450.51, < 12.556.0a0",
     "psutil >= 5.6.6",
     "cachetools >= 1.0.1",
     "termcolor >= 1.0.0",
diff --git a/requirements.txt b/requirements.txt
index 2170099..c54f477 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 # Sync with pyproject.toml and nvitop/version.py
-nvidia-ml-py >= 11.450.51, < 12.551.0a0
+nvidia-ml-py >= 11.450.51, < 12.556.0a0
 psutil >= 5.6.6
 cachetools >= 1.0.1
 termcolor >= 1.0.0