[cli] add missing requirement (#805)

2022-04-19 13:56:59 +08:00 · 2022-04-19 13:56:59 +08:00 · 05d9ae5999
parent de2f581d43
commit 05d9ae5999
3 changed files with 5 additions and 1 deletions
--- a/colossalai/cli/init.py
+++ b/colossalai/cli/init.py
@ -0,0 +1,3 @@
+from .cli import cli
+
+__all__ = ['cli']
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@ -7,3 +7,4 @@ tensorboard
 packaging
 pre-commit
 rich
+click
--- a/setup.py
+++ b/setup.py
@ -215,7 +215,7 @@ setup(
    install_requires=fetch_requirements('requirements/requirements.txt'),
    entry_points='''
        [console_scripts]
-        colossal=colossalai.cli.cli:cli
+        colossal=colossalai.cli:cli
    ''',
    python_requires='>=3.7',
    classifiers=[