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)