From 307841379693c023494d021a3d42b046f853eb96 Mon Sep 17 00:00:00 2001
From: Georg Reich <georgreich99@gmail.com>
Date: Tue, 13 Feb 2024 18:13:14 +0100
Subject: [PATCH] added jax install to workflow

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

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 299eec71..5c4652ff 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -26,6 +26,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip setuptools
           pip install flake8 codecov pytest-cov wheel matplotlib seaborn mne
+          pip install jax[cpu]==0.3.25 -f https://storage.googleapis.com/jax-releases/jax_releases.html
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
           pip install .
       - name: Lint with flake8 🎓