From 2a72d9b17804ce4770186c4d4bb18a92cbb05a36 Mon Sep 17 00:00:00 2001 From: "mulin.lyh" Date: Wed, 3 Jul 2024 19:18:49 +0800 Subject: [PATCH] fix logging --- modelscope/utils/ast_utils.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/modelscope/utils/ast_utils.py b/modelscope/utils/ast_utils.py index 3775748d..657ebb33 100644 --- a/modelscope/utils/ast_utils.py +++ b/modelscope/utils/ast_utils.py @@ -21,7 +21,6 @@ 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 p = Path(__file__) @@ -58,10 +57,8 @@ 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) + ast_logger.setLevel(logging.INFO) return ast_logger @@ -722,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: @@ -732,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(