cleanup

Signed-off-by: Mayank Mishra <[email protected]>
IBM · Oct 24, 2024 · f2a84c2 · f2a84c2
1 parent a4f6b55
commit f2a84c2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/dolomite_engine/utils/logger.py b/dolomite_engine/utils/logger.py
@@ -31,6 +31,11 @@ def get_logger() -> logging.Logger:
 
 @run_rank_n
 def log_rank_0(level: int, msg: str) -> None:
+    logger = get_logger()
+    if logger is None:
+        set_logger()
+        log_rank_0(logging.WARN, "logger is not initialized yet, initializing now")
+
     get_logger().log(level=level, msg=msg, stacklevel=3)