cache Python dependencies for CI (#1623)

NanoComp · Jun 21, 2021 · 8903b50 · 8903b50
1 parent ff405c6
commit 8903b50
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 2 deletions.
diff --git a/.github/workflows/build-ci.yml b/.github/workflows/build-ci.yml
@@ -108,11 +108,26 @@ jobs:
       with:
         python-version: ${{ matrix.python-version }}
 
+    - name: Get pip cache dir
+      id: pip-cache
+      run: |
+        python -m pip install --upgrade pip wheel
+        echo "::set-output name=dir::$(pip cache dir)"
+
+    - name: pip cache
+      uses: actions/cache@v2
+      id: cache
+      with:
+        path: ${{ steps.pip-cache.outputs.dir }}
+        key: ${{ runner.os }}-py${{ matrix.python-version }}-mpi-${{ matrix.enable-mpi }}-pip-${{ hashFiles('**/requirements.txt') }}
+        restore-keys: ${{ runner.os }}-py${{ matrix.python-version }}-mpi-${{ matrix.enable-mpi }}-pip-
+
     - name: Install Python dependencies
-      run: pip install autograd h5py jax jaxlib matplotlib numpy parameterized pytest scipy
+      if: steps.cache.outputs.cache-hit != 'true'
+      run: pip install -r python/requirements.txt
 
     - name: Install mpi4py
-      if: matrix.enable-mpi
+      if: ${{ matrix.enable-mpi && steps.cache.outputs.cache-hit != true }}
       run: pip install mpi4py
 
     - name: Run autoreconf

diff --git a/python/requirements.txt b/python/requirements.txt
@@ -0,0 +1,9 @@
+autograd
+h5py
+jax
+jaxlib
+matplotlib
+numpy
+parameterized
+pytest
+scipy