fix 'print log' in cli

2 years ago · 9b5655f6ad
parent 02681b72b3
commit 9b5655f6ad
1 changed files with 1 additions and 1 deletions
--- a/paddlespeech/s2t/utils/log.py
+++ b/paddlespeech/s2t/utils/log.py
@ -99,7 +99,7 @@ class Log():
    _call_from_cli = False
    _frame = inspect.currentframe()
    while _frame:
-        if 'paddlespeech/cli/__init__.py' in _frame.f_code.co_filename or 'paddlespeech/t2s' in _frame.f_code.co_filename:
+        if 'paddlespeech/cli/entry.py' in _frame.f_code.co_filename or 'paddlespeech/t2s' in _frame.f_code.co_filename:
            _call_from_cli = True
            break
        _frame = _frame.f_back