1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-02 21:48:15 +00:00
ColossalAI/colossalai/cli/check/__init__.py
Hongxin Liu 079bf3cb26
[misc] update pre-commit and run all files ()
* [misc] update pre-commit

* [misc] run pre-commit

* [misc] remove useless configuration files

* [misc] ignore cuda for clang-format
2023-09-19 14:20:26 +08:00

15 lines
396 B
Python

import click
from .check_installation import check_installation
__all__ = ["check"]
@click.command(help="Check if Colossal-AI is correct based on the given option")
@click.option("-i", "--installation", is_flag=True, help="Check if Colossal-AI is built correctly")
def check(installation):
if installation:
check_installation()
return
click.echo("No option is given")