diff --git a/.github/workflows/test_python.yml b/.github/workflows/test_python.yml
index e735dc315a..7a67fb23b6 100644
--- 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
index 41b4e048a7..73066e6ca8 100644
--- a/setup.py
+++ b/setup.py
@@ -83,6 +83,7 @@
         "deepmd/infer",
         "deepmd/loss",
         "deepmd/utils",
+        "deepmd/cluster"
     ],
     python_requires=">=3.6",
     classifiers=[