mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-04-28 11:45:23 +00:00
079bf3cb26
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
23 lines
310 B
Python
23 lines
310 B
Python
import click
|
|
|
|
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)
|
|
|
|
if __name__ == "__main__":
|
|
cli()
|