Skip to content

Commit 16aa5ad

Browse files
committed
misc: Linting
1 parent 98d6328 commit 16aa5ad

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

tests/test_gpu_common.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,8 +130,16 @@ def test_no_visible_devices(self, env_variables):
130130
def test_visible_devices_uuid(self):
131131
# Query GPU 0's UUID independently of _get_uuid_to_index_map
132132
probes = [
133-
(['nvidia-smi', '-L'], r'GPU\s+0:.*\(UUID:\s*([\w-]+)\)', 'CUDA_VISIBLE_DEVICES'),
134-
(['rocm-smi', '--showuniqueid'], r'GPU\[0\].*Unique ID:\s*([\w]+)', 'ROCR_VISIBLE_DEVICES'),
133+
(
134+
['nvidia-smi', '-L'],
135+
r'GPU\s+0:.*\(UUID:\s*([\w-]+)\)',
136+
'CUDA_VISIBLE_DEVICES'
137+
),
138+
(
139+
['rocm-smi', '--showuniqueid'],
140+
r'GPU\[0\].*Unique ID:\s*([\w]+)',
141+
'ROCR_VISIBLE_DEVICES'
142+
),
135143
]
136144
uuid = env_var = None
137145
for cmd, pattern, var in probes:

0 commit comments

Comments
 (0)