File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments