Skip to content

Commit 1fbe63e

Browse files
committed
Fix tests
1 parent 3966af9 commit 1fbe63e

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

cuda_core/cuda/core/system/_process.pxi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ class ProcessInfo:
3333
Only valid for processes running on MIG devices.
3434
"""
3535
if not self._device.mig.is_mig_device:
36-
raise nvml.UnsupportedError(
36+
raise nvml.NotSupportedError(
3737
"GPU instance ID is only valid for processes running on MIG devices."
3838
)
3939
return self._process_info.gpu_instance_id
@@ -46,7 +46,7 @@ class ProcessInfo:
4646
Only valid for processes running on MIG devices.
4747
"""
4848
if not self._device.mig.is_mig_device:
49-
raise nvml.UnsupportedError(
49+
raise nvml.NotSupportedError(
5050
"Compute instance ID is only valid for processes running on MIG devices."
5151
)
5252
return self._process_info.compute_instance_id

cuda_core/tests/system/test_system_device.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -744,9 +744,9 @@ def test_compute_running_processes():
744744
assert isinstance(proc.gpu_instance_id, int)
745745
assert isinstance(proc.compute_instance_id, int)
746746
else:
747-
with pytest.raises(nvml.UnsupportedError):
747+
with pytest.raises(nvml.NotSupportedError):
748748
proc.gpu_instance_id # noqa: B018
749-
with pytest.raises(nvml.UnsupportedError):
749+
with pytest.raises(nvml.NotSupportedError):
750750
proc.compute_instance_id # noqa: B018
751751

752752

0 commit comments

Comments
 (0)