def decide_accelerator_type()

in src/backend/main.py [0:0]

• 13 lines of code
• 6 McCabe index (conditional complexity)

def decide_accelerator_type(machine_type):
    if machine_type.startswith("a3-highgpu"):
        return "NVIDIA_H100_80GB"
    elif machine_type.startswith("a2-ultragpu"):
        return "NVIDIA_A100_80GB"
    elif machine_type.startswith("a2-highgpu"):
        return "NVIDIA_TESLA_A100"
    elif machine_type.startswith("g"):
        return "NVIDIA_L4"
    elif machine_type.startswith("n"):
        return "NVIDIA_TESLA_T4"
    else:
        return "NVIDIA_L4"