From e6200d4680934cd24e9a8e714c9746339a51fbe1 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Mon, 16 Jan 2023 12:56:30 -0500
Subject: [PATCH] prevent logging messages from other modules

Fix #2238.

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
---
 deepmd/loggers/loggers.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/deepmd/loggers/loggers.py b/deepmd/loggers/loggers.py
index 3bb9e9fa4c..04321486ad 100644
--- a/deepmd/loggers/loggers.py
+++ b/deepmd/loggers/loggers.py
@@ -192,12 +192,10 @@ def set_log_handles(
     os.environ['TF_CPP_MIN_LOG_LEVEL'] = str(int((level / 10) - 1))
 
     # get root logger
-    root_log = logging.getLogger()
+    root_log = logging.getLogger("deepmd")
+    root_log.propagate = False
 
-    # remove all old handlers
     root_log.setLevel(level)
-    for hdlr in root_log.handlers[:]:
-        root_log.removeHandler(hdlr)
 
     # check if arguments are present
     MPI = None