diff --git a/whisperx/alignment.py b/whisperx/alignment.py index 765cb82a..7637683d 100644 --- a/whisperx/alignment.py +++ b/whisperx/alignment.py @@ -15,7 +15,7 @@ from whisperx.audio import SAMPLE_RATE, load_audio from whisperx.utils import interpolate_nans -from whisperx.types import ( +from whisperx.schema import ( AlignedTranscriptionResult, SingleSegment, SingleAlignedSegment, diff --git a/whisperx/asr.py b/whisperx/asr.py index 67d8a8be..1ad3408f 100644 --- a/whisperx/asr.py +++ b/whisperx/asr.py @@ -12,7 +12,7 @@ from transformers.pipelines.pt_utils import PipelineIterator from whisperx.audio import N_SAMPLES, SAMPLE_RATE, load_audio, log_mel_spectrogram -from whisperx.types import SingleSegment, TranscriptionResult +from whisperx.schema import SingleSegment, TranscriptionResult from whisperx.vads import Vad, Silero, Pyannote diff --git a/whisperx/diarize.py b/whisperx/diarize.py index 3632cd8c..6bab799d 100644 --- a/whisperx/diarize.py +++ b/whisperx/diarize.py @@ -5,7 +5,7 @@ import torch from whisperx.audio import load_audio, SAMPLE_RATE -from whisperx.types import TranscriptionResult, AlignedTranscriptionResult +from whisperx.schema import TranscriptionResult, AlignedTranscriptionResult class DiarizationPipeline: diff --git a/whisperx/types.py b/whisperx/schema.py similarity index 100% rename from whisperx/types.py rename to whisperx/schema.py diff --git a/whisperx/transcribe.py b/whisperx/transcribe.py index 8eca427a..0b94c134 100644 --- a/whisperx/transcribe.py +++ b/whisperx/transcribe.py @@ -10,7 +10,7 @@ from whisperx.asr import load_model from whisperx.audio import load_audio from whisperx.diarize import DiarizationPipeline, assign_word_speakers -from whisperx.types import AlignedTranscriptionResult, TranscriptionResult +from whisperx.schema import AlignedTranscriptionResult, TranscriptionResult from whisperx.utils import LANGUAGES, TO_LANGUAGE_CODE, get_writer