Module neuralops.utils.logger

Functions

def get_logger(name: str = 'neuralops') ‑> logging.Logger

Expand source code

def get_logger(name: str = "neuralops") -> logging.Logger:
    """Return the logger."""
    return logging.getLogger(name)

Return the logger.

def set_log_level(logger_name: str, level: int) ‑> None

Expand source code

def set_log_level(logger_name: str, level: int) -> None:
    """Set the log level."""
    logging.getLogger(logger_name).setLevel(level)

Set the log level.

def setup_logger(name: str,
level: int = 20,
log_file: str | None = None,
format_string: str | None = None) ‑> logging.Logger

Expand source code

def setup_logger(
    name: str,
    level: int = logging.INFO,
    log_file: str | None = None,
    format_string: str | None = None,
) -> logging.Logger:
    """Setup logger."""
    if format_string is None:
        format_string = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
    logger = logging.getLogger(name)
    logger.setLevel(level)
    logger.handlers.clear()
    console = logging.StreamHandler(sys.stdout)
    console.setFormatter(logging.Formatter(format_string))
    logger.addHandler(console)
    if log_file:
        Path(log_file).parent.mkdir(parents=True, exist_ok=True)
        file_handler = logging.FileHandler(log_file)
        file_handler.setFormatter(logging.Formatter(format_string))
        logger.addHandler(file_handler)
    return logger

Setup logger.