From 9035cf7cd0291c86df664e9fbfc34e302897c492 Mon Sep 17 00:00:00 2001
From: "mulin.lyh" <mulin.lyh@taobao.com>
Date: Wed, 3 Jul 2024 19:32:19 +0800
Subject: [PATCH] fix logger issue

---
 modelscope/utils/ast_utils.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/modelscope/utils/ast_utils.py b/modelscope/utils/ast_utils.py
index 3554b0ea..657ebb33 100644
--- a/modelscope/utils/ast_utils.py
+++ b/modelscope/utils/ast_utils.py
@@ -21,10 +21,8 @@ from modelscope.metainfo import (CustomDatasets, Heads, Hooks, LR_Schedulers,
                                  Preprocessors, TaskModels, Trainers)
 from modelscope.utils.constant import Fields, Tasks
 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 +56,15 @@ TEMPLATE_PATH = 'TEMPLATE_PATH'
 TEMPLATE_FILE = 'ast_index_file.py'
 
 
+def get_ast_logger():
+    ast_logger = logging.getLogger('modelscope.ast')
+    ast_logger.setLevel(logging.INFO)
+    return ast_logger
+
+
+logger = get_ast_logger()
+
+
 class AstScanning(object):
 
     def __init__(self) -> None:
@@ -712,7 +719,6 @@ def load_index(
     cache_dir = os.getenv('MODELSCOPE_CACHE', indexer_file_dir)
     index_file = os.getenv('MODELSCOPE_INDEX_FILE', indexer_file)
     file_path = os.path.join(cache_dir, index_file)
-    logger.info(f'Loading ast index from {file_path}')
     index = None
 
     if force_rebuild:
@@ -722,6 +728,7 @@ def load_index(
 
     # when developing, we need to generator as need.
     if __is_develop_model():
+        logger.info(f'Loading ast index from {file_path}')
         if os.path.exists(file_path):  # already exist, check it's latest
             wrapped_index = _load_index(file_path)
             md5, files_mtime = file_scanner.files_mtime_md5(