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