clear log handlers before setup (#1603)

This commit is contained in:
Yifan Yang 2024-04-24 14:31:25 +08:00 committed by GitHub
parent 9f8f0bceb5
commit 368b7d10a7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -110,6 +110,13 @@ def str2bool(v):
raise argparse.ArgumentTypeError("Boolean value expected.") raise argparse.ArgumentTypeError("Boolean value expected.")
def clear_log_handlers():
logger = logging.getLogger()
handlers = logger.handlers[:]
for handler in handlers:
logger.removeHandler(handler)
def setup_logger( def setup_logger(
log_filename: Pathlike, log_filename: Pathlike,
log_level: str = "info", log_level: str = "info",
@ -126,6 +133,8 @@ def setup_logger(
use_console: use_console:
True to also print logs to console. True to also print logs to console.
""" """
clear_log_handlers()
now = datetime.now() now = datetime.now()
date_time = now.strftime("%Y-%m-%d-%H-%M-%S") date_time = now.strftime("%Y-%m-%d-%H-%M-%S")
if dist.is_available() and dist.is_initialized(): if dist.is_available() and dist.is_initialized():