diff --git a/paddlespeech/s2t/utils/log.py b/paddlespeech/s2t/utils/log.py index 4f51b7f05e5c14b4ec7877b32ee9cc016ee5ba13..b711ab7396f928f6d7f0963bd9ae20d4a07e83d9 100644 --- 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