From 7f05f8dd41dbab01f09c480a9128b7adc1b2e2ca Mon Sep 17 00:00:00 2001
From: Mirko Bunse <mirko.bunse@cs.tu-dortmund.de>
Date: Mon, 1 Jul 2024 18:10:29 +0200
Subject: [PATCH] Fix the autodoc of the composable module

---
 .github/workflows/ci.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index fb0647b..fb5e8c7 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -43,7 +43,7 @@ jobs:
       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
+          python -m pip install --upgrade pip setuptools wheel "jax[cpu]"
           python -m pip install -e .[composable,neural,docs]
         docs-folder: "docs/"
     - name: Publish documentation