From 13d8a82ec7a7997b62439088ab0029d6244c6434 Mon Sep 17 00:00:00 2001 From: danielnugraha Date: Thu, 5 Oct 2023 10:19:09 +0200 Subject: [PATCH 1/2] Update jax requirements --- examples/quickstart-jax/pyproject.toml | 11 +++++------ examples/quickstart-jax/requirements.txt | 8 ++++---- 2 files changed, 9 insertions(+), 10 deletions(-) diff --git a/examples/quickstart-jax/pyproject.toml b/examples/quickstart-jax/pyproject.toml index 6a67cff6f4b..7653bfb31d0 100644 --- a/examples/quickstart-jax/pyproject.toml +++ b/examples/quickstart-jax/pyproject.toml @@ -5,12 +5,11 @@ description = "JAX example training a linear regression model with federated lea authors = ["The Flower Authors "] [tool.poetry.dependencies] -python = "^3.8" -flwr = "^1.0.0" -jax = "^0.4.0" -jaxlib = "^0.4.0" -scikit-learn = "^1.1.1" -numpy = "^1.21.4" +python = ">=3.8,<3.11" +flwr = "1.0.0" +jax = "0.4.13" +jaxlib = "0.4.13" +scikit-learn = "1.1.1" [build-system] requires = ["poetry-core>=1.4.0"] diff --git a/examples/quickstart-jax/requirements.txt b/examples/quickstart-jax/requirements.txt index bf7a9c64d66..2caf88cceee 100644 --- a/examples/quickstart-jax/requirements.txt +++ b/examples/quickstart-jax/requirements.txt @@ -1,4 +1,4 @@ -flwr~=1.4.0 -jax~=0.4.10 -numpy~=1.21.1 -scikit_learn~=1.2.2 +flwr>=1.0,<2.0 +jax==0.4.13 +jaxlib==0.4.13 +scikit-learn==1.1.1 From 18097c24083c4441db3156c7d4f7b055d7897d48 Mon Sep 17 00:00:00 2001 From: danielnugraha Date: Thu, 5 Oct 2023 10:21:58 +0200 Subject: [PATCH 2/2] Use recent jax version --- examples/quickstart-jax/pyproject.toml | 4 ++-- examples/quickstart-jax/requirements.txt | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/examples/quickstart-jax/pyproject.toml b/examples/quickstart-jax/pyproject.toml index 7653bfb31d0..41b4462d0a1 100644 --- a/examples/quickstart-jax/pyproject.toml +++ b/examples/quickstart-jax/pyproject.toml @@ -7,8 +7,8 @@ authors = ["The Flower Authors "] [tool.poetry.dependencies] python = ">=3.8,<3.11" flwr = "1.0.0" -jax = "0.4.13" -jaxlib = "0.4.13" +jax = "0.4.17" +jaxlib = "0.4.17" scikit-learn = "1.1.1" [build-system] diff --git a/examples/quickstart-jax/requirements.txt b/examples/quickstart-jax/requirements.txt index 2caf88cceee..964f07a51b7 100644 --- a/examples/quickstart-jax/requirements.txt +++ b/examples/quickstart-jax/requirements.txt @@ -1,4 +1,4 @@ flwr>=1.0,<2.0 -jax==0.4.13 -jaxlib==0.4.13 +jax==0.4.17 +jaxlib==0.4.17 scikit-learn==1.1.1