nextrec 0.4.1__py3-none-any.whl → 0.4.3__py3-none-any.whl

nextrec/basic/loggers.py

@@ -13,40 +13,42 @@ import json
 import copy
 import logging
 import numbers
+
 from typing import Mapping, Any
 from nextrec.basic.session import create_session, Session
 
 ANSI_CODES = {
-    'black': '\033[30m',
-    'red': '\033[31m',
-    'green': '\033[32m',
-    'yellow': '\033[33m',
-    'blue': '\033[34m',
-    'magenta': '\033[35m',
-    'cyan': '\033[36m',
-    'white': '\033[37m',
-    'bright_black': '\033[90m',
-    'bright_red': '\033[91m',
-    'bright_green': '\033[92m',
-    'bright_yellow': '\033[93m',
-    'bright_blue': '\033[94m',
-    'bright_magenta': '\033[95m',
-    'bright_cyan': '\033[96m',
-    'bright_white': '\033[97m',
+    "black": "\033[30m",
+    "red": "\033[31m",
+    "green": "\033[32m",
+    "yellow": "\033[33m",
+    "blue": "\033[34m",
+    "magenta": "\033[35m",
+    "cyan": "\033[36m",
+    "white": "\033[37m",
+    "bright_black": "\033[90m",
+    "bright_red": "\033[91m",
+    "bright_green": "\033[92m",
+    "bright_yellow": "\033[93m",
+    "bright_blue": "\033[94m",
+    "bright_magenta": "\033[95m",
+    "bright_cyan": "\033[96m",
+    "bright_white": "\033[97m",
 }
 
-ANSI_BOLD = '\033[1m'
-ANSI_RESET = '\033[0m'
-ANSI_ESCAPE_PATTERN = re.compile(r'\033\[[0-9;]*m')
+ANSI_BOLD = "\033[1m"
+ANSI_RESET = "\033[0m"
+ANSI_ESCAPE_PATTERN = re.compile(r"\033\[[0-9;]*m")
 
 DEFAULT_LEVEL_COLORS = {
-    'DEBUG': 'cyan',
-    'INFO': None,
-    'WARNING': 'yellow',
-    'ERROR': 'red',
-    'CRITICAL': 'bright_red',
+    "DEBUG": "cyan",
+    "INFO": None,
+    "WARNING": "yellow",
+    "ERROR": "red",
+    "CRITICAL": "bright_red",
 }
 
+
 class AnsiFormatter(logging.Formatter):
     def __init__(
         self,
@@ -65,16 +67,17 @@ class AnsiFormatter(logging.Formatter):
         record_copy = copy.copy(record)
         formatted = super().format(record_copy)
 
-        if self.auto_color_level and '\033[' not in formatted:
+        if self.auto_color_level and "\033[" not in formatted:
             color = self.level_colors.get(record.levelname)
             if color:
                 formatted = colorize(formatted, color=color)
 
         if self.strip_ansi:
-            return ANSI_ESCAPE_PATTERN.sub('', formatted)
+            return ANSI_ESCAPE_PATTERN.sub("", formatted)
 
         return formatted
 
+
 def colorize(text: str, color: str | None = None, bold: bool = False) -> str:
     """Apply ANSI color and bold formatting to the given text."""
     if not color and not bold:
@@ -87,43 +90,53 @@ def colorize(text: str, color: str | None = None, bold: bool = False) -> str:
     result += text + ANSI_RESET
     return result
 
+
 def setup_logger(session_id: str | os.PathLike | None = None):
     """Set up a logger that logs to both console and a file with ANSI formatting.
-       Only console output has colors; file output is stripped of ANSI codes.
-       Logs are stored under ``log/<experiment_id>/logs`` by default. A stable
-       log file is used per experiment so multiple components (e.g. data
-       processor and model training) append to the same file instead of creating
-       separate timestamped files.
+    Only console output has colors; file output is stripped of ANSI codes.
+    Logs are stored under ``log/<experiment_id>/logs`` by default. A stable
+    log file is used per experiment so multiple components (e.g. data
+    processor and model training) append to the same file instead of creating
+    separate timestamped files.
     """
 
     session = create_session(str(session_id) if session_id is not None else None)
     log_dir = session.logs_dir
     log_dir.mkdir(parents=True, exist_ok=True)
-    log_file = log_dir / f"{session.log_basename}.log"
+    log_file = log_dir / "runs.log"
+
+    console_format = "%(message)s"
+    file_format = "%(asctime)s - %(levelname)s - %(message)s"
+    date_format = "%Y-%m-%d %H:%M:%S"
 
-    console_format = '%(message)s'
-    file_format = '%(asctime)s - %(levelname)s - %(message)s'
-    date_format = '%Y-%m-%d %H:%M:%S'
-    
     logger = logging.getLogger()
     logger.setLevel(logging.INFO)
 
     if logger.hasHandlers():
         logger.handlers.clear()
 
-    file_handler = logging.FileHandler(log_file, encoding='utf-8')
+    file_handler = logging.FileHandler(log_file, encoding="utf-8")
     file_handler.setLevel(logging.INFO)
-    file_handler.setFormatter(AnsiFormatter(file_format, datefmt=date_format, strip_ansi=True))
+    file_handler.setFormatter(
+        AnsiFormatter(file_format, datefmt=date_format, strip_ansi=True)
+    )
 
     console_handler = logging.StreamHandler(sys.stdout)
     console_handler.setLevel(logging.INFO)
-    console_handler.setFormatter(AnsiFormatter(console_format, datefmt=date_format, auto_color_level=True,))
+    console_handler.setFormatter(
+        AnsiFormatter(
+            console_format,
+            datefmt=date_format,
+            auto_color_level=True,
+        )
+    )
 
     logger.addHandler(file_handler)
     logger.addHandler(console_handler)
-    
+
     return logger
 
+
 class TrainingLogger:
     def __init__(
         self,
@@ -146,7 +159,9 @@ class TrainingLogger:
         try:
             from torch.utils.tensorboard import SummaryWriter  # type: ignore
         except ImportError:
-            logging.warning("[TrainingLogger] tensorboard not installed, disable tensorboard logging.")
+            logging.warning(
+                "[TrainingLogger] tensorboard not installed, disable tensorboard logging."
+            )
             self.enable_tensorboard = False
             return
         tb_dir = self.session.logs_dir / "tensorboard"
@@ -158,7 +173,9 @@ class TrainingLogger:
     def tensorboard_logdir(self):
         return self.tb_dir
 
-    def format_metrics(self, metrics: Mapping[str, Any], split: str) -> dict[str, float]:
+    def format_metrics(
+        self, metrics: Mapping[str, Any], split: str
+    ) -> dict[str, float]:
         formatted: dict[str, float] = {}
         for key, value in metrics.items():
             if isinstance(value, numbers.Number):
@@ -170,7 +187,9 @@ class TrainingLogger:
                     continue
         return formatted
 
-    def log_metrics(self, metrics: Mapping[str, Any], step: int, split: str = "train") -> None:
+    def log_metrics(
+        self, metrics: Mapping[str, Any], step: int, split: str = "train"
+    ) -> None:
         payload = self.format_metrics(metrics, split)
         payload["step"] = int(step)
         with self.log_path.open("a", encoding="utf-8") as f:
@@ -188,4 +207,4 @@ class TrainingLogger:
         if self.tb_writer:
             self.tb_writer.flush()
             self.tb_writer.close()
-            self.tb_writer = None
+            self.tb_writer = None