def get_num_efa_devices()

in launcher/nemo/stages.py [0:0]

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

def get_num_efa_devices(instance_type):
    # If not a EFA instance, return 0
    if instance_type not in efa_supported_instance:
        return 0
    # If multi-EFA, return from mapping
    if instance_type in instanceWithMultipleEFAs:
        return instanceWithMultipleEFAs[instance_type]
    # Only a single EFA device
    return 1