Pin jax to be <0.4.15 also in environment.yml

ami-iit · Mar 1, 2024 · 4819b72 · 4819b72
1 parent d0f4d4e
commit 4819b72
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/environment.yml b/environment.yml
@@ -5,8 +5,8 @@ dependencies:
   # Dependencies from setup.cfg
   - python=3.11
   - coloredlogs
-  - jax >= 0.4.13
-  - jaxlib
+  - jax >= 0.4.13,<0.4.15
+  - jaxlib < 0.4.15
   - jaxlie >= 1.3.0
   - jax-dataclasses >= 1.4.0
   - pptree