Skip to content

Commit 6f61894

Browse files
committed
remove import pyaudio and dependencies in the beginning to keep simuleval lightweight
1 parent 7d771b5 commit 6f61894

File tree

1 file changed

+13
-4
lines changed

1 file changed

+13
-4
lines changed

simuleval/evaluator/remote.py

+13-4
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,16 @@
99
import threading
1010
import time
1111
from queue import Queue
12-
13-
import wave
1412
import numpy as np
15-
import pyaudio
16-
from silero_vad import load_silero_vad, read_audio, get_speech_timestamps
13+
14+
try:
15+
import wave
16+
import pyaudio
17+
from silero_vad import load_silero_vad, read_audio, get_speech_timestamps
18+
except:
19+
wave, pyaudio, load_silero_vad, read_audio, get_speech_timestamps = [
20+
None for _ in range(5)
21+
]
1722

1823
from simuleval.data.segments import (
1924
Segment,
@@ -71,6 +76,10 @@ def remote_eval(self):
7176

7277
class DemoRemote(RemoteEvaluator):
7378
def __init__(self, evaluator: SentenceLevelEvaluator) -> None:
79+
if None in [wave, pyaudio, load_silero_vad, read_audio, get_speech_timestamps]:
80+
raise Exception(
81+
"Please install wave, pyaudio, and silero_vad to run the demo"
82+
)
7483
super().__init__(evaluator)
7584
self.float_array = np.asarray([])
7685
self.sample_rate = 16000

0 commit comments

Comments
 (0)