From 51b34edbfeb64153da9bf7ada16adc726f566842 Mon Sep 17 00:00:00 2001 From: Ivan Zaitsev Date: Wed, 17 Sep 2025 16:53:58 -0700 Subject: [PATCH] [autorevert] Configure root logger for AWS lambda --- .../pytorch_auto_revert/__main__.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/aws/lambda/pytorch-auto-revert/pytorch_auto_revert/__main__.py b/aws/lambda/pytorch-auto-revert/pytorch_auto_revert/__main__.py index a3d6bf0c3f..bcf6cf3f8e 100644 --- a/aws/lambda/pytorch-auto-revert/pytorch_auto_revert/__main__.py +++ b/aws/lambda/pytorch-auto-revert/pytorch_auto_revert/__main__.py @@ -23,7 +23,22 @@ def setup_logging(log_level: str) -> None: numeric_level = getattr(logging, log_level.upper(), None) if not isinstance(numeric_level, int): raise ValueError(f"Invalid log level: {log_level}") - logging.basicConfig(level=numeric_level) + + root_logger = logging.getLogger() + root_logger.setLevel(numeric_level) + + if not root_logger.handlers: + handler = logging.StreamHandler() + handler.setLevel(numeric_level) + formatter = logging.Formatter( + "%(asctime)s %(levelname)s [%(name)s] %(message)s" + ) + handler.setFormatter(formatter) + root_logger.addHandler(handler) + else: + for handler in root_logger.handlers: + if handler.level == logging.NOTSET: + handler.setLevel(numeric_level) def get_opts() -> argparse.Namespace: