Class Logger

datastudio/utils/logging.py:63–340

Configurable logger with file, console, and WandB backends. Example config:: logger = dict( type='Logger', level='INFO', log_file='logs/run.log', console=True, wandb=dict(enabled=True, project='datastudio', name='exp_1'),

@HOOKS.register_module()
class Logger:
    """Configurable logger with file, console, and WandB backends.

    Example config::

        logger = dict(
            type='Logger',
            level='INFO',
            log_file='logs/run.log',
            console=True,
            wandb=dict(enabled=True, project='datastudio', name='exp_1'),
        )
    """

    _instance: Optional["Logger"] = None

    def __init__(
        self,
        name: str = "datastudio",
        level: Union[str, int] = "INFO",
        log_file: Optional[str] = None,
        console: bool = True,
        colored: bool = True,
        work_dir: Optional[str] = None,
        wandb_config: Optional[Dict] = None,
        timestamp: bool = True,
        **kwargs,
    ):
        self.name = name
        self._logger = logging.getLogger(name)
        self._logger.setLevel(
            getattr(logging, level.upper()) if isinstance(level, str) else level
        )
        self._logger.handlers.clear()

        self.work_dir = work_dir or os.getcwd()
        self._wandb_enabled = False

        # Console handler
        if console:
            console_handler = logging.StreamHandler(sys.stdout)
            console_handler.setLevel(self._logger.level)
            fmt = "%(asctime)s %(levelname)s %(message)s"
            datefmt = "%Y-%m-%d %H:%M:%S"
            formatter = ColoredFormatter(fmt, datefmt, use_color=colored)
            console_handler.setFormatter(formatter)
            self._logger.addHandler(console_handler)

        # File handler
        if log_file:
            # Add timestamp to log filename if enabled
            if timestamp:
                log_file = self._add_timestamp_to_filename(log_file)

            log_path = (
                log_file
                if os.path.isabs(log_file)
                else os.path.join(self.work_dir, log_file)