feat(framework) Allow graceful exit of start_server via a single `C…

…trl+C` (#4846)
adap · Jan 21, 2025 · 2527881 · 2527881
1 parent f5b3253
commit 2527881
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/py/flwr/server/app.py b/src/py/flwr/server/app.py
@@ -228,6 +228,13 @@ def start_server(  # pylint: disable=too-many-arguments,too-many-locals
         "enabled" if certificates is not None else "disabled",
     )
 
+    # Graceful shutdown
+    register_exit_handlers(
+        event_type=EventType.START_SERVER_LEAVE,
+        exit_message="Flower server terminated gracefully.",
+        grpc_servers=[grpc_server],
+    )
+
     # Start training
     hist = run_fl(
         server=initialized_server,