feat(framework) Include default number of server rounds in templates

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

@@ -6,12 +6,15 @@ from flwr.server import ServerApp, ServerAppComponents, ServerConfig
 
 
 def server_fn(context: Context):
+ # Read from config
+ num_rounds = int(context.run_config["num-server-rounds"])
+
  # Define strategy
  strategy = FedAvg(
  fraction_fit=1.0,
  fraction_evaluate=1.0,
  )
- config = ServerConfig(num_rounds=3)
+ config = ServerConfig(num_rounds=num_rounds)
 
  return ServerAppComponents(strategy=strategy, config=config)
 

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

@@ -6,9 +6,12 @@ from flwr.server import ServerApp, ServerAppComponents, ServerConfig
 
 
 def server_fn(context: Context):
+ # Read from config
+ num_rounds = int(context.run_config["num-server-rounds"])
+
  # Define strategy
  strategy = FedAvg()
- config = ServerConfig(num_rounds=3)
+ config = ServerConfig(num_rounds=num_rounds)
 
  return ServerAppComponents(strategy=strategy, config=config)
 

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

@@ -6,9 +6,12 @@ from flwr.server.strategy import FedAvg
 
 
 def server_fn(context: Context):
+ # Read from config
+ num_rounds = int(context.run_config["num-server-rounds"])
+
  # Define strategy
  strategy = FedAvg()
- config = ServerConfig(num_rounds=3)
+ config = ServerConfig(num_rounds=num_rounds)
 
  return ServerAppComponents(strategy=strategy, config=config)
 

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

@@ -6,9 +6,12 @@ from flwr.server.strategy import FedAvg
 
 
 def server_fn(context: Context):
+ # Read from config
+ num_rounds = int(context.run_config["num-server-rounds"])
+
  # Define strategy
  strategy = FedAvg()
- config = ServerConfig(num_rounds=3)
+ config = ServerConfig(num_rounds=num_rounds)
 
  return ServerAppComponents(strategy=strategy, config=config)
 

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

@@ -12,14 +12,17 @@ ndarrays = get_weights(Net())
 parameters = ndarrays_to_parameters(ndarrays)
 
 def server_fn(context: Context):
+ # Read from config
+ num_rounds = int(context.run_config["num-server-rounds"])
+
  # Define strategy
  strategy = FedAvg(
  fraction_fit=1.0,
  fraction_evaluate=1.0,
  min_available_clients=2,
  initial_parameters=parameters,
  )
- config = ServerConfig(num_rounds=3)
+ config = ServerConfig(num_rounds=num_rounds)
 
  return ServerAppComponents(strategy=strategy, config=config)
 

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

@@ -6,13 +6,16 @@ from flwr.server.strategy import FedAvg
 
 
 def server_fn(context: Context):
+ # Read from config
+ num_rounds = int(context.run_config["num-server-rounds"])
+
  # Define strategy
  strategy = FedAvg(
  fraction_fit=1.0,
  fraction_evaluate=1.0,
  min_available_clients=2,
  )
- config = ServerConfig(num_rounds=3)
+ config = ServerConfig(num_rounds=num_rounds)
 
  return ServerAppComponents(strategy=strategy, config=config)
 

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

@@ -12,14 +12,17 @@ config = ServerConfig(num_rounds=3)
 parameters = ndarrays_to_parameters(load_model().get_weights())
 
 def server_fn(context: Context):
+ # Read from config
+ num_rounds = int(context.run_config["num-server-rounds"])
+
  # Define strategy
  strategy = strategy = FedAvg(
  fraction_fit=1.0,
  fraction_evaluate=1.0,
  min_available_clients=2,
  initial_parameters=parameters,
  )
- config = ServerConfig(num_rounds=3)
+ config = ServerConfig(num_rounds=num_rounds)
 
  return ServerAppComponents(strategy=strategy, config=config)
 

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

@@ -29,6 +29,9 @@ publisher = "$username"
 serverapp = "$import_name.app:server"
 clientapp = "$import_name.app:client"
 
+[tool.flwr.app.config]
+num-server-rounds = "3"
+
 [tool.flwr.federations]
 default = "localhost"
 

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

@@ -27,6 +27,9 @@ publisher = "$username"
 serverapp = "$import_name.server:app"
 clientapp = "$import_name.client:app"
 
+[tool.flwr.app.config]
+num-server-rounds = "3"
+
 [tool.flwr.federations]
 default = "localhost"
 

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

@@ -24,6 +24,9 @@ publisher = "$username"
 serverapp = "$import_name.server:app"
 clientapp = "$import_name.client:app"
 
+[tool.flwr.app.config]
+num-server-rounds = "3"
+
 [tool.flwr.federations]
 default = "localhost"
 

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

@@ -24,6 +24,9 @@ publisher = "$username"
 serverapp = "$import_name.server:app"
 clientapp = "$import_name.client:app"
 
+[tool.flwr.app.config]
+num-server-rounds = "3"
+
 [tool.flwr.federations]
 default = "localhost"
 

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

@@ -22,6 +22,9 @@ publisher = "$username"
 serverapp = "$import_name.server:app"
 clientapp = "$import_name.client:app"
 
+[tool.flwr.app.config]
+num-server-rounds = "3"
+
 [tool.flwr.federations]
 default = "localhost"
 

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

@@ -24,6 +24,9 @@ publisher = "$username"
 serverapp = "$import_name.server:app"
 clientapp = "$import_name.client:app"
 
+[tool.flwr.app.config]
+num-server-rounds = "3"
+
 [tool.flwr.federations]
 default = "localhost"
 

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

@@ -23,6 +23,9 @@ publisher = "$username"
 serverapp = "$import_name.server:app"
 clientapp = "$import_name.client:app"
 
+[tool.flwr.app.config]
+num-server-rounds = "3"
+
 [tool.flwr.federations]
 default = "localhost"
 

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

@@ -23,6 +23,9 @@ publisher = "$username"
 serverapp = "$import_name.server:app"
 clientapp = "$import_name.client:app"
 
+[tool.flwr.app.config]
+num-server-rounds = "3"
+
 [tool.flwr.federations]
 default = "localhost"