mogen/apis/__init__.py

from mogen.apis import test, train
from mogen.apis.test import (
    collect_results_cpu,
    collect_results_gpu,
    multi_gpu_test,
    single_gpu_test,
)
from mogen.apis.train import set_random_seed, train_model

__all__ = [
    'collect_results_cpu', 'collect_results_gpu', 'multi_gpu_test',
    'single_gpu_test', 'set_random_seed', 'train_model'
]