From da766cf3adc77be479651d56b5b0fa6121620950 Mon Sep 17 00:00:00 2001
From: Charles Beauville <charles@flower.ai>
Date: Sun, 21 Jul 2024 22:12:04 +0200
Subject: [PATCH] fix(framework:skip) Fix flwr new templates config (#3863)

---
 src/py/flwr/cli/new/templates/app/pyproject.hf.toml.tpl  | 8 ++++----
 src/py/flwr/cli/new/templates/app/pyproject.jax.toml.tpl | 6 +++---
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/py/flwr/cli/new/templates/app/pyproject.hf.toml.tpl b/src/py/flwr/cli/new/templates/app/pyproject.hf.toml.tpl
index 7b7ea9d01200..a15455a3b225 100644
--- a/src/py/flwr/cli/new/templates/app/pyproject.hf.toml.tpl
+++ b/src/py/flwr/cli/new/templates/app/pyproject.hf.toml.tpl
@@ -11,10 +11,10 @@ dependencies = [
     "flwr[simulation]>=1.9.0,<2.0",
     "flwr-datasets>=0.0.2,<1.0.0",
     "torch==2.2.1",
-    "transformers>=4.30.0,<5.0"
-    "evaluate>=0.4.0,<1.0"
-    "datasets>=2.0.0, <3.0"
-    "scikit-learn>=1.3.1, <2.0"
+    "transformers>=4.30.0,<5.0",
+    "evaluate>=0.4.0,<1.0",
+    "datasets>=2.0.0, <3.0",
+    "scikit-learn>=1.3.1, <2.0",
 ]
 
 [tool.hatch.build.targets.wheel]
diff --git a/src/py/flwr/cli/new/templates/app/pyproject.jax.toml.tpl b/src/py/flwr/cli/new/templates/app/pyproject.jax.toml.tpl
index e899f48f4c5c..c0cce3e19d06 100644
--- a/src/py/flwr/cli/new/templates/app/pyproject.jax.toml.tpl
+++ b/src/py/flwr/cli/new/templates/app/pyproject.jax.toml.tpl
@@ -9,9 +9,9 @@ description = ""
 license = "Apache-2.0"
 dependencies = [
     "flwr[simulation]>=1.9.0,<2.0",
-    "jax==0.4.26",
-    "jaxlib==0.4.26",
-    "scikit-learn==1.4.2",
+    "jax==0.4.13",
+    "jaxlib==0.4.13",
+    "scikit-learn==1.3.2",
 ]
 
 [tool.hatch.build.targets.wheel]