AirTerminalDevice/main.py

from .apps.train.train import train

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