From b264da5b171c45222969ad7bc06eb675dbccee64 Mon Sep 17 00:00:00 2001
From: Wassim Kabalan <wastondev@gmail.com>
Date: Sun, 8 Dec 2024 23:13:31 +0100
Subject: [PATCH] JAX version is 0.4.35 until Diffrax new release

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

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 2c2299d..12baaf7 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -29,7 +29,7 @@ jobs:
       run: |
         sudo apt-get install -y libopenmpi-dev
         python -m pip install --upgrade pip
-        pip install jax=0.4.35
+        pip install jax==0.4.35
         pip install .[test]
 
     - name: Run Single Device Tests