AirTerminalDevice/main.py

def main(*args,config=None):
    mode = config['mode']
    equp_type = config['equp_type']
    
    if equp_type == 'DHU':
        if mode == 'optimize':
            from .apps.DHU.optimize import optimize
            optimize(*args,config=config)
        elif mode == 'train':
            from .apps.DHU.train import train
            train(*args,config=config)
        else:
            raise NotImplementedError(mode)
    else:
        raise NotImplementedError(equp_type)