From 50b3b171c1541301f1731e771c0ada3e536c4479 Mon Sep 17 00:00:00 2001 From: YuanTingHsieh Date: Thu, 26 Sep 2024 15:20:10 -0700 Subject: [PATCH] Update flwr server --- .../hello-flower/flwr-pt-tb/flwr_pt_tb/server.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/examples/hello-world/hello-flower/flwr-pt-tb/flwr_pt_tb/server.py b/examples/hello-world/hello-flower/flwr-pt-tb/flwr_pt_tb/server.py index 3093d3b5e8..e0936b8d54 100644 --- a/examples/hello-world/hello-flower/flwr-pt-tb/flwr_pt_tb/server.py +++ b/examples/hello-world/hello-flower/flwr-pt-tb/flwr_pt_tb/server.py @@ -14,7 +14,8 @@ from typing import List, Tuple from flwr.common import Metrics, ndarrays_to_parameters -from flwr.server import ServerApp, ServerConfig +from flwr.server import ServerApp, ServerConfig, ServerAppComponents +from flwr.common import Context from flwr.server.strategy import FedAvg from .task import Net, get_weights @@ -53,13 +54,16 @@ def weighted_average(metrics: List[Tuple[int, Metrics]]) -> Metrics: initial_parameters=parameters, ) - # Define config config = ServerConfig(num_rounds=3) # Flower ServerApp -app = ServerApp( - config=config, - strategy=strategy, -) +def server_fn(context: Context): + return ServerAppComponents( + strategy=strategy, + config=config, + ) + + +app = ServerApp(server_fn=server_fn)