AirTerminalDevice/main.py

from .apps.DHU.train import train as train_DHU

def main(*args,config=None):
    mode = config['mode']
    equp_type = config['equp_type']
    
    if mode == 'train' and equp_type == 'DHU':
        train_DHU(*args,config=config)
    else:
        raise NotImplementedError