mirror of https://github.com/hpcaitech/ColossalAI
15 lines
396 B
Python
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")
|