Remove file handler entirely if logfile not used.

Signed-off-by: Robin Kupper <[email protected]>
mlflow · May 2, 2023 · 62e1f94 · 62e1f94
1 parent a05cd65
commit 62e1f94
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mlflow_export_import/common/logging_utils.py b/mlflow_export_import/common/logging_utils.py
@@ -52,7 +52,8 @@ def _create_default_log_config(output_path=None, log_format=None):
         file_handler = cfg["handlers"]["file"]
         file_handler["filename"] = output_path
     else:
-        handlers = cfg["root"]["handlers"]
-        handlers.remove("file")
+        # Remove file handler
+        del cfg["handlers"]["file"]
+        del cfg["root"]["handlers"]["file"]
 
     return cfg