From 50fbe0c532d847f2ec0fbc1ae32e82fb4ce0923a Mon Sep 17 00:00:00 2001
From: ThibeauWouters <thibeauwouters@gmail.com>
Date: Thu, 19 Dec 2024 18:07:55 +0100
Subject: [PATCH] Pin jax version and update v0.0.2

---
 setup.cfg | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/setup.cfg b/setup.cfg
index c2d4a9e..8190562 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = fiestaEM
-version = 0.0.1
+version = 0.0.2
 author = Thibeau Wouters
 author_email = thibeauwouters@gmail.com
 url = https://github.com/ThibeauWouters/fiestaEM
@@ -15,8 +15,8 @@ packages_dir=
     =src
 packages = find:
 install_requires =
-    jax>=0.4.24
-    jaxlib>=0.4.24
+    jax<=0.4.31
+    jaxlib<=0.4.31
     numpy<2.0.0
     pandas<2.0.0
     jaxtyping