From b824621e6b9be3e0c32290d490176575740b0b39 Mon Sep 17 00:00:00 2001 From: HAOCHENYE <21724054@zju.edu.cn> Date: Fri, 10 Mar 2023 17:53:36 +0800 Subject: [PATCH] Fix as comment --- mmengine/logging/logger.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/mmengine/logging/logger.py b/mmengine/logging/logger.py index ba445955c8..94ca1a7955 100644 --- a/mmengine/logging/logger.py +++ b/mmengine/logging/logger.py @@ -11,7 +11,7 @@ from mmengine.utils.manager import _accquire_lock, _release_lock -class UniqueWarningFilter(logging.Filter): +class FilterDuplicateWarning(logging.Filter): """Filter the repeated warning message. Args: @@ -34,9 +34,8 @@ def filter(self, record: LogRecord) -> bool: if record.levelno != logging.WARNING: return True - msg = record.msg - if msg not in self.seen: - self.seen.add(msg) + if record.msg not in self.seen: + self.seen.add(record.msg) return True return False @@ -194,7 +193,7 @@ def __init__(self, # Only rank0 `StreamHandler` will log messages below error level. stream_handler.setLevel(log_level) if rank == 0 else \ stream_handler.setLevel(logging.ERROR) - stream_handler.addFilter(UniqueWarningFilter(logger_name)) + stream_handler.addFilter(FilterDuplicateWarning(logger_name)) self.handlers.append(stream_handler) if log_file is not None: @@ -222,7 +221,7 @@ def __init__(self, file_handler.setFormatter( MMFormatter(color=False, datefmt='%Y/%m/%d %H:%M:%S')) file_handler.setLevel(log_level) - file_handler.addFilter(UniqueWarningFilter(logger_name)) + file_handler.addFilter(FilterDuplicateWarning(logger_name)) self.handlers.append(file_handler) @classmethod