chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix: remove DiarizationPipeline from public API

Browse Source
This commit is contained in:
Barabazs
2025-05-02 13:04:24 +02:00
parent 7d36b832f9
commit 36d552cad3
2 changed files with 35 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
whisperx/__init__.py
View File

@ -29,12 +29,3 @@ def load_audio(*args, **kwargs):
def assign_word_speakers(*args, **kwargs):
diarize = _lazy_import("diarize")
return diarize.assign_word_speakers(*args, **kwargs)
class DiarizationPipeline:
def __init__(self, *args, **kwargs):
diarize = _lazy_import("diarize")
self._pipeline = diarize.DiarizationPipeline(*args, **kwargs)
def __getattr__(self, name):
return getattr(self._pipeline, name)