deps(nvidia-ml-py): add nvidia-ml-py 12.535.133 to support list

XuehaiPan · Nov 4, 2023 · 27cf25e · 27cf25e
1 parent 5cba62f
commit 27cf25e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -25,7 +25,7 @@ repos:
       - id: debug-statements
       - id: double-quote-string-fixer
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.1.2
+    rev: v0.1.4
     hooks:
       - id: ruff
         args: [--fix, --exit-non-zero-on-fix]

diff --git a/nvitop/api/device.py b/nvitop/api/device.py
@@ -2814,16 +2814,16 @@ def parse_cuda_visible_devices(
         >>> import os
         >>> os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'
         >>> os.environ['CUDA_VISIBLE_DEVICES'] = '6,5'
-        >>> parse_cuda_visible_devices()        # parse the `CUDA_VISIBLE_DEVICES` environment variable to NVML indices
+        >>> parse_cuda_visible_devices()       # parse the `CUDA_VISIBLE_DEVICES` environment variable to NVML indices
         [6, 5]
 
-        >>> parse_cuda_visible_devices('0,4')     # pass the `CUDA_VISIBLE_DEVICES` value explicitly
+        >>> parse_cuda_visible_devices('0,4')  # pass the `CUDA_VISIBLE_DEVICES` value explicitly
         [0, 4]
 
         >>> parse_cuda_visible_devices('GPU-18ef14e9,GPU-849d5a8d')  # accept abbreviated UUIDs
         [5, 6]
 
-        >>> parse_cuda_visible_devices(None)    # get all devices when the `CUDA_VISIBLE_DEVICES` environment variable unset
+        >>> parse_cuda_visible_devices(None)   # get all devices when the `CUDA_VISIBLE_DEVICES` environment variable unset
         [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
 
         >>> parse_cuda_visible_devices('MIG-d184f67c-c95f-5ef2-a935-195bd0094fbd')           # MIG device support (MIG UUID)
@@ -2833,11 +2833,11 @@ def parse_cuda_visible_devices(
         >>> parse_cuda_visible_devices('MIG-GPU-3eb79704/13/0')                              # MIG device support (abbreviated GPU UUID)
         [(0, 1)]
 
-        >>> parse_cuda_visible_devices('')      # empty string
+        >>> parse_cuda_visible_devices('')     # empty string
         []
-        >>> parse_cuda_visible_devices('0,0')   # invalid `CUDA_VISIBLE_DEVICES` (duplicate device ordinal)
+        >>> parse_cuda_visible_devices('0,0')  # invalid `CUDA_VISIBLE_DEVICES` (duplicate device ordinal)
         []
-        >>> parse_cuda_visible_devices('16')    # invalid `CUDA_VISIBLE_DEVICES` (device ordinal out of range)
+        >>> parse_cuda_visible_devices('16')   # invalid `CUDA_VISIBLE_DEVICES` (device ordinal out of range)
         []
     """  # pylint: disable=line-too-long
     if cuda_visible_devices is _VALUE_OMITTED:

diff --git a/nvitop/version.py b/nvitop/version.py
@@ -71,6 +71,7 @@
     '11.525.131',
     '12.535.77',
     '12.535.108',
+    '12.535.133',
 )
 """The list of supported ``nvidia-ml-py`` versions.
 See also: `nvidia-ml-py's Release History <https://pypi.org/project/nvidia-ml-py/#history>`_.