Fix version of Jax to solve Poetry resolve issue (#2709)

adap · Dec 17, 2023 · 06d08d5 · 06d08d5
1 parent 316bbdb
commit 06d08d5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/e2e/jax/pyproject.toml b/e2e/jax/pyproject.toml
@@ -7,8 +7,8 @@ authors = ["The Flower Authors <[email protected]>"]
 [tool.poetry.dependencies]
 python = "^3.8"
 flwr = { path = "../../", develop = true, extras = ["simulation"] }
-jax = "^0.4.0"
-jaxlib = "^0.4.0"
+jax = "==0.4.13"
+jaxlib = "==0.4.13"
 scikit-learn = "^1.1.1"
 numpy = "^1.21.4"