From 7c985299702d7a9e4d4739afaec652cca422cec3 Mon Sep 17 00:00:00 2001 From: Javier Date: Sun, 19 May 2024 07:05:51 +0100 Subject: [PATCH] option to mute everything ran by a ray backend (#3435) --- .../flwr/server/superlink/fleet/vce/backend/raybackend.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/py/flwr/server/superlink/fleet/vce/backend/raybackend.py b/src/py/flwr/server/superlink/fleet/vce/backend/raybackend.py index 265ce55d2c3e..93aca583af9c 100644 --- a/src/py/flwr/server/superlink/fleet/vce/backend/raybackend.py +++ b/src/py/flwr/server/superlink/fleet/vce/backend/raybackend.py @@ -55,7 +55,12 @@ def __init__( runtime_env = ( self._configure_runtime_env(work_dir=work_dir) if work_dir else None ) - if backend_config.get("silent", False): + + if backend_config.get("mute_logging", False): + init_ray( + logging_level=WARNING, log_to_driver=False, runtime_env=runtime_env + ) + elif backend_config.get("silent", False): init_ray(logging_level=WARNING, log_to_driver=True, runtime_env=runtime_env) else: init_ray(runtime_env=runtime_env)