option to mute everything ran by a ray backend (#3435)

adap · May 19, 2024 · 7c98529 · 7c98529
1 parent 13bac6a
commit 7c98529
Showing 1 changed file with 6 additions and 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
@@ -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)