Skip to content

Commit b74b642

Browse files
committed
fix(caul): allow partial imports for inference runners
1 parent b67be75 commit b74b642

2 files changed

Lines changed: 22 additions & 7 deletions

File tree

caul-core/caul_core/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from .asr_pipeline import ASRPipeline, InferenceRunner
2-
from .asr_task import Postprocessor, Preprocessor
1+
from .asr_pipeline import ASRPipeline
2+
from .asr_task import Postprocessor, Preprocessor, InferenceRunner
33
from .config import (
44
ASRPipelineConfig,
55
BasePreprocessorConfig,
Lines changed: 20 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,20 @@
1-
from .parakeet import ParakeetInferenceRunner
2-
from .faster_whisper import FasterWhisperInferenceRunner
3-
from .fireredasr import FireRedASR2InferenceRunner
4-
from .parakeet_trt import ParakeetTrtInferenceRunner
5-
from .whisper_trt import WhisperTrtInferenceRunner
1+
try:
2+
from .parakeet import ParakeetInferenceRunner
3+
except ModuleNotFoundError:
4+
ParakeetInferenceRunner = None
5+
try:
6+
from .faster_whisper import FasterWhisperInferenceRunner
7+
except ModuleNotFoundError:
8+
FasterWhisperInferenceRunner = None
9+
try:
10+
from .fireredasr import FireRedASR2InferenceRunner
11+
except ModuleNotFoundError:
12+
FireRedASR2InferenceRunner = None
13+
try:
14+
from .parakeet_trt import ParakeetTrtInferenceRunner
15+
except ModuleNotFoundError:
16+
ParakeetTrtInferenceRunner = None
17+
try:
18+
from .whisper_trt import WhisperTrtInferenceRunner
19+
except ModuleNotFoundError:
20+
WhisperTrtInferenceRunner = None

0 commit comments

Comments
 (0)