fix cluster subpackage not included

njzjz · Feb 11, 2021 · 6958e1a · 6958e1a
1 parent be7b2fa
commit 6958e1a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/.github/workflows/test_python.yml b/.github/workflows/test_python.yml
@@ -54,6 +54,14 @@ jobs:
     - uses: actions/setup-python@v2
       with:
         python-version: ${{ matrix.python }}
+    - name: pip cache
+      uses: actions/cache@v2
+      with:
+        path: ~/.cache/pip
+        key:
+          ${{ runner.os }}-pip-${{ hashFiles('**/setup.py') }}-${{ hashFiles('**/requirements.txt') }}
+        restore-keys: |
+          ${{ runner.os }}-pip-
     - run: |
         sudo apt update
         sudo apt install gcc-${{ matrix.gcc }} g++-${{ matrix.gcc }}

diff --git a/setup.py b/setup.py
@@ -83,6 +83,7 @@
         "deepmd/infer",
         "deepmd/loss",
         "deepmd/utils",
+        "deepmd/cluster"
     ],
     python_requires=">=3.6",
     classifiers=[