[cli] add missing requirement (#805)

05d9ae59 · Frank Lee · GitHub · de2f581d · 05d9ae59 · 05d9ae59
Unverified Commit 05d9ae59 authored Apr 19, 2022 by Frank Lee Committed by GitHub Apr 19, 2022
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

colossalai/cli/__init__.py colossalai/cli/__init__.py +3 -0

requirements/requirements.txt requirements/requirements.txt +1 -0

setup.py setup.py +1 -1

No files found.
--- a/colossalai/cli/__init__.py
+++ b/colossalai/cli/__init__.py
+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=[