Support from different logging message per threads/processes

ecmwf · Dec 5, 2024 · 92f20bf · 92f20bf
1 parent 84b16b0
commit 92f20bf
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/anemoi/utils/logs.py b/src/anemoi/utils/logs.py
@@ -14,14 +14,13 @@
 import threading
 
 thread_local = threading.local()
-thread_local.logging_name = "main"
 
 
 LOGGER = logging.getLogger(__name__)
 
 
-def set_logging_name(logging_name):
-    thread_local.logging_name = logging_name
+def set_logging_name(name):
+    thread_local.logging_name = name
 
 
 class ThreadCustomFormatter(logging.Formatter):
@@ -30,7 +29,8 @@ def format(self, record):
         return super().format(record)
 
 
-def enable_logging_name(logging_name):
+def enable_logging_name(name="main"):
+    thread_local.logging_name = name
 
     formatter = ThreadCustomFormatter("%(asctime)s - %(logging_name)s - %(levelname)s - %(message)s")