Skip to content

Commit 943aa96

Browse files
chore: show pyaudio install message (#121)
1 parent f89906f commit 943aa96

File tree

2 files changed

+13
-4
lines changed

2 files changed

+13
-4
lines changed

scripts/asr/transcribe_mic.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,12 @@
66
import riva.client
77
from riva.client.argparse_utils import add_asr_config_argparse_parameters, add_connection_argparse_parameters
88

9-
import riva.client.audio_io
10-
9+
try:
10+
import riva.client.audio_io
11+
except ModuleNotFoundError as e:
12+
print(f"ModuleNotFoundError: {e}")
13+
print("Please install pyaudio from https://pypi.org/project/PyAudio")
14+
exit(1)
1115

1216
def parse_args() -> argparse.Namespace:
1317
default_device_info = riva.client.audio_io.get_default_input_device_info()

scripts/tts/talk.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,8 +70,13 @@ def parse_args() -> argparse.Namespace:
7070
args = parser.parse_args()
7171
if args.output is not None:
7272
args.output = args.output.expanduser()
73-
if args.list_devices or args.output_device or args.play_audio:
74-
import riva.client.audio_io
73+
try:
74+
if args.list_devices or args.output_device or args.play_audio:
75+
import riva.client.audio_io
76+
except ModuleNotFoundError as e:
77+
print(f"ModuleNotFoundError: {e}")
78+
print("Please install pyaudio from https://pypi.org/project/PyAudio")
79+
exit(1)
7580
return args
7681

7782

0 commit comments

Comments
 (0)