fix(framework:cli): Fix tensorflow template (#3380)

src/py/flwr/cli/new/templates/app/code/client.tensorflow.py.tpl

@@ -1,14 +1,9 @@
 """$project_name: A Flower / TensorFlow app."""
 
-import os
-
 from flwr.client import NumPyClient, ClientApp
 
-from $project_name.task import load_data, load_model
-
+from $import_name.task import load_data, load_model
 
-# Make TensorFlow log less verbose
-os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
 
 # Define Flower Client and client_fn
 class FlowerClient(NumPyClient):
@@ -43,4 +38,6 @@ def client_fn(cid):
 
 
 # Flower ClientApp
-app = ClientApp(client_fn)
+app = ClientApp(
+    client_fn=client_fn,
+)

src/py/flwr/cli/new/templates/app/code/server.tensorflow.py.tpl

@@ -4,7 +4,10 @@ from flwr.common import ndarrays_to_parameters
 from flwr.server import ServerApp, ServerConfig
 from flwr.server.strategy import FedAvg
 
-from $project_name.task import load_model
+from $import_name.task import load_model
+
+# Define config
+config = ServerConfig(num_rounds=3)
 
 parameters = ndarrays_to_parameters(load_model().get_weights())
 
@@ -19,6 +22,6 @@ strategy = FedAvg(
 
 # Create ServerApp
 app = ServerApp(
-    config=ServerConfig(num_rounds=3),
+    config=config,
     strategy=strategy,
 )

src/py/flwr/cli/new/templates/app/pyproject.numpy.toml.tpl

@@ -9,7 +9,7 @@ description = ""
 authors = [
     { name = "The Flower Authors", email = "[email protected]" },
 ]
-license = {text = "Apache License (2.0)"}
+license = { text = "Apache License (2.0)" }
 dependencies = [
     "flwr[simulation]>=1.8.0,<2.0",
     "numpy>=1.21.0",

src/py/flwr/cli/new/templates/app/pyproject.pytorch.toml.tpl

@@ -9,7 +9,7 @@ description = ""
 authors = [
     { name = "The Flower Authors", email = "[email protected]" },
 ]
-license = {text = "Apache License (2.0)"}
+license = { text = "Apache License (2.0)" }
 dependencies = [
     "flwr[simulation]>=1.8.0,<2.0",
     "flwr-datasets[vision]>=0.0.2,<1.0.0",

src/py/flwr/cli/new/templates/app/pyproject.sklearn.toml.tpl

@@ -9,7 +9,7 @@ description = ""
 authors = [
     { name = "The Flower Authors", email = "[email protected]" },
 ]
-license = {text = "Apache License (2.0)"}
+license = { text = "Apache License (2.0)" }
 dependencies = [
     "flwr[simulation]>=1.8.0,<2.0",
     "flwr-datasets[vision]>=0.0.2,<1.0.0",

src/py/flwr/cli/new/templates/app/pyproject.tensorflow.toml.tpl

@@ -9,7 +9,7 @@ description = ""
 authors = [
     { name = "The Flower Authors", email = "[email protected]" },
 ]
-license = {text = "Apache License (2.0)"}
+license = { text = "Apache License (2.0)" }
 dependencies = [
     "flwr[simulation]>=1.8.0,<2.0",
     "flwr-datasets[vision]>=0.0.2,<1.0.0",