def get_num_gpus()

in utils/aml_common.py [0:0]

• 9 lines of code
• 1 McCabe index (conditional complexity)

def get_num_gpus(azure_compute_cluster_size):
    num_gpu_dict = {
        "Standard_NC24ads_A100_v4": 1,
        "Standard_NC48ads_A100_v4": 2,
        "Standard_NC96ads_A100_v4": 4,
        "Standard_NC40ads_H100_v5": 1,
        "Standard_NC80adis_H100_v5": 2    
    }
    return num_gpu_dict[azure_compute_cluster_size]