Spaces:
Running
Running
File size: 1,841 Bytes
9098220 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | from __future__ import annotations
import logging
import os
import warnings
class LightningNoiseFilter(logging.Filter):
"""Suppress non-actionable Lightning promotional messages."""
_BLOCKED_SUBSTRINGS = (
"for seamless cloud logging",
"for seamless cloud uploads",
"try installing [litlogger]",
"try installing [litmodels]",
)
def filter(self, record: logging.LogRecord) -> bool:
msg = record.getMessage().lower()
return not any(substring in msg for substring in self._BLOCKED_SUBSTRINGS)
def suppress_lightning_noise() -> None:
"""
Reduce noisy Lightning advice/promotional logs while keeping real warnings/errors.
"""
os.environ.setdefault("DISABLE_LIGHTNING_ADVICE", "1")
for logger_name in (
"lightning",
"lightning.pytorch",
"lightning.pytorch.utilities.rank_zero",
"pytorch_lightning",
):
lg = logging.getLogger(logger_name)
lg.setLevel(logging.WARNING)
if not any(isinstance(f, LightningNoiseFilter) for f in lg.filters):
lg.addFilter(LightningNoiseFilter())
warnings.filterwarnings(
"ignore",
message=r".*try installing \[litlogger\].*",
)
warnings.filterwarnings(
"ignore",
message=r".*try installing \[litmodels\].*",
)
def configure_cli_logging(level: int = logging.INFO) -> None:
"""Configure project CLI logs and reduce third-party INFO chatter."""
logging.basicConfig(
level=level,
format="%(asctime)s %(levelname)s %(name)s %(message)s",
)
for logger_name in (
"lightning",
"lightning.pytorch",
"lightning.pytorch.utilities.rank_zero",
"pytorch_lightning",
"torch",
):
logging.getLogger(logger_name).setLevel(logging.WARNING)
|