fix log issue

This commit is contained in:
mulin.lyh
2024-07-03 19:00:31 +08:00
parent 4e189eec54
commit 696da5c272

View File

@@ -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: