From 1e999c27e94ffc275aee486d131a75ec8073e85d Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Mon, 25 Apr 2022 03:11:54 +0000 Subject: [PATCH] fix exit, test=doc --- paddlespeech/cli/asr/infer.py | 4 ++-- tests/unit/cli/test_cli.sh | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/paddlespeech/cli/asr/infer.py b/paddlespeech/cli/asr/infer.py index 050d6f7f..0fb54868 100644 --- a/paddlespeech/cli/asr/infer.py +++ b/paddlespeech/cli/asr/infer.py @@ -346,7 +346,7 @@ class ASRExecutor(BaseExecutor): max_duration = 50.0 if audio_duration >= max_duration: logger.error("Please input audio file less then 50 seconds.\n") - exit(1) + return False except Exception as e: logger.exception(e) logger.error( @@ -383,7 +383,7 @@ class ASRExecutor(BaseExecutor): ) == "n" or content.strip() == "no" or content.strip( ) == "No": logger.info("Exit the program") - exit(1) + return False else: logger.warning("Not regular input, please input again") diff --git a/tests/unit/cli/test_cli.sh b/tests/unit/cli/test_cli.sh index c00e5e29..926b1ac0 100755 --- a/tests/unit/cli/test_cli.sh +++ b/tests/unit/cli/test_cli.sh @@ -22,7 +22,7 @@ paddlespeech asr --model deepspeech2offline_librispeech --lang en --input ./en.w # long audio restriction wget -c wget https://paddlespeech.bj.bcebos.com/datasets/single_wav/zh/test_long_audio_01.wav paddlespeech asr --input test_long_audio_01.wav -if [ $? -ne 1 ]; then +if [ $? -ne -1 ]; then exit 1 fi -- GitLab