27 lines
646 B
Python
27 lines
646 B
Python
import logging
|
|
import argparse
|
|
|
|
from trainlib.cli import train
|
|
|
|
logger: logging.Logger = logging.getLogger(__name__)
|
|
|
|
|
|
def create_parser() -> argparse.ArgumentParser:
|
|
parser = argparse.ArgumentParser(
|
|
description="trainlib cli",
|
|
# formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
parser.add_argument(
|
|
"--log-level",
|
|
type=int,
|
|
metavar="int",
|
|
choices=[10, 20, 30, 40, 50],
|
|
help="Log level: 10=DEBUG, 20=INFO, 30=WARNING, 40=ERROR, 50=CRITICAL",
|
|
)
|
|
|
|
subparsers = parser.add_subparsers(help="subcommand help")
|
|
train.register_parser(subparsers)
|
|
|
|
return parser
|
|
|