def __init__()

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

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

    def __init__(self, cfg):
        super().__init__(cfg)
        # Use GPU device for default flow for NeMo runs
        self.device = "gpu"
        self.instance_type = get_instance_type(cfg)
        self.num_efa_devices = get_num_efa_devices(self.instance_type)
        self.telemetry = Telemetry()