def get_device_name()

in src/optimum/nvidia/utils/nvml.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

def get_device_name(device: int) -> Optional[Tuple[str, ...]]:
    try:
        nvml_device_handle = nvmlDeviceGetHandleByIndex(device)
        device_full_name = nvmlDeviceGetName(nvml_device_handle)
        return device_full_name.split()
    except NVMLError:
        return None