githubmirror/ColossalAI

colossalai/cli/cli.py

25 строк373 B

Wangbo Zhao(黑色枷锁)

[NFC] polish colossalai/cli/cli.py code style (#2734)

15 фев 2023, 17:25

15 фев 2023, 17:258331420

import click

from .benchmark import benchmark
from .check import check
from .launcher import run


class Arguments():

    def __init__(self, arg_dict):
        for k, v in arg_dict.items():
            self.__dict__[k] = v


@click.group()
def cli():
    pass


cli.add_command(run)
cli.add_command(check)
cli.add_command(benchmark)

if __name__ == '__main__':
    cli()