Pin jax version and update v0.0.2

ThibeauWouters · Dec 19, 2024 · 50fbe0c · 50fbe0c
1 parent 187c291
commit 50fbe0c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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 = [email protected]
 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