From 696da5c272fb0fb396f34ea5c8d2fb84c614326b Mon Sep 17 00:00:00 2001
From: "mulin.lyh" <mulin.lyh@taobao.com>
Date: Wed, 3 Jul 2024 19:00:31 +0800
Subject: [PATCH] fix log issue

---
 modelscope/utils/ast_utils.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/modelscope/utils/ast_utils.py b/modelscope/utils/ast_utils.py
index 3554b0ea..3775748d 100644
--- a/modelscope/utils/ast_utils.py
+++ b/modelscope/utils/ast_utils.py
@@ -24,7 +24,6 @@ from modelscope.utils.file_utils import get_modelscope_cache_dir
 from modelscope.utils.logger import get_logger
 from modelscope.utils.registry import default_group
 
-logger = get_logger(log_level=logging.WARNING)
 p = Path(__file__)
 
 # get the path of package 'modelscope'
@@ -58,6 +57,17 @@ TEMPLATE_PATH = 'TEMPLATE_PATH'
 TEMPLATE_FILE = 'ast_index_file.py'
 
 
+def get_ast_logger():
+    _logger = get_logger()
+    ast_logger = logging.getLogger('modelscope.ast')
+    for handler in _logger.handlers:
+        ast_logger.addHandler(handler)
+    return ast_logger
+
+
+logger = get_ast_logger()
+
+
 class AstScanning(object):
 
     def __init__(self) -> None: