diff --git a/colossalai/cli/cli.py b/colossalai/cli/cli.py index 3e5b9ae63..a94e1150e 100644 --- a/colossalai/cli/cli.py +++ b/colossalai/cli/cli.py @@ -1,7 +1,8 @@ import click -from .launcher import run -from .check import check + from .benchmark import benchmark +from .check import check +from .launcher import run class Arguments():