diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index c040676..3c48dc1 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -40,15 +40,16 @@ jobs:
     if: github.ref == 'refs/heads/master'
     steps:
       - uses: actions/checkout@v1
-
+      - name: Install dependencies
+        run: |
+          pip install --no-cache-dir jax[cpu] scikit-learn pandas tqdm matplotlib joblib xlrd abstention ucimlrepo certifi
+          pip install --no-cache-dir -e .[composable,neural,docs]
       - name: Build documentation
         uses: ammaraskar/sphinx-action@master
         with:
           pre-build-command: |
             apt-get --allow-releaseinfo-change update -y && apt-get install -y git && git --version
             python -m pip install --upgrade pip setuptools wheel
-            pip install --no-cache-dir jax[cpu] scikit-learn pandas tqdm matplotlib joblib xlrd abstention ucimlrepo certifi
-            pip install --no-cache-dir -e .[composable,neural,docs]
           docs-folder: "docs/"
 
       - name: Publish documentation