maint: fix jax version specifier

Gattocrucco · Mar 27, 2023 · 0153b52 · 0153b52
1 parent 2b1934f
commit 0153b52
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -50,8 +50,8 @@
     install_requires=[
         'numpy>=1.20', # first version with broadcast_shapes
         'scipy>=1.5', # minimum required by jax
-        'jax>=0.4.1<=0.4.6', # 0.4.1 first supporting type dispatching on jax arrays
-        'jaxlib>=0.4.1<=0.4.6', # 0.4.6 last supporting old jit implementation, the new one triggers a leaked tracer exception in linalg operations
+        'jax>=0.4.1,<=0.4.6', # 0.4.1 first supporting type dispatching on jax arrays
+        'jaxlib>=0.4.1,<=0.4.6', # 0.4.6 last supporting old jit implementation, the new one triggers a leaked tracer exception in linalg operations
         'gvar>=11.10.1', # first supporting gvar.gvar(x, cov) with readonly cov
     ]
 )