config_cli.py 414 Bytes
Newer Older
Jinjing Zhou's avatar
Jinjing Zhou committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from ..pipeline import *
from ..utils.factory import ModelFactory, PipelineFactory
import typer
from enum import Enum
import typing
import yaml
from pathlib import Path

config_app = typer.Typer(help="Generate the config files")
for key, pipeline in PipelineFactory.registry.items():
    config_app.command(key, help=pipeline.get_description())(pipeline.get_cfg_func())

if __name__ == "__main__":
    config_app()