github-starred/nvitop
2
0
Fork 0
mirror of https://github.com/XuehaiPan/nvitop.git synced 2026-05-21 06:45:24 -06:00
Code Issues 29 Projects Packages Wiki Activity Actions

chore(api/device): update type annotations

Browse source
This commit is contained in:
Xuehai Pan 2023-05-01 03:36:35 +00:00
parent 6697d54895
commit c730af2106
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
nvitop/api/device.py
View file

@ -2141,7 +2141,9 @@ class CudaDevice(Device):
If the index is out of range for the given ``CUDA_VISIBLE_DEVICES`` environment variable.
""" # pylint: disable=line-too-long
_nvml_index: int | tuple[int, int]
_nvml_index: int
index: int
nvml_index: int
@classmethod
def is_available(cls) -> bool:
@ -2317,9 +2319,9 @@ Device.cuda = CudaDevice
class CudaMigDevice(CudaDevice, MigDevice): # type: ignore[misc]
"""Class for CUDA devices that are MIG devices."""
_nvml_index: tuple[int, int]
index: tuple[int, int]
nvml_index: tuple[int, int]
_nvml_index: tuple[int, int] # type: ignore[assignment]
index: tuple[int, int] # type: ignore[assignment]
nvml_index: tuple[int, int] # type: ignore[assignment]
def is_mig_device_uuid(uuid: str | None) -> bool: