From 3b1b9a8c4d2c7874f9733779d0b74a136b0964cd Mon Sep 17 00:00:00 2001 From: Barabazs <31799121+Barabazs@users.noreply.github.com> Date: Thu, 9 Oct 2025 12:22:58 +0000 Subject: [PATCH] refactor: rename types.py to schema.py to avoid stdlib conflict --- whisperx/alignment.py | 2 +- whisperx/asr.py | 2 +- whisperx/diarize.py | 2 +- whisperx/{types.py => schema.py} | 0 whisperx/transcribe.py | 2 +- 5 files changed, 4 insertions(+), 4 deletions(-) rename whisperx/{types.py => schema.py} (100%) diff --git a/whisperx/alignment.py b/whisperx/alignment.py index 765cb82ad..7637683d3 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 67d8a8be2..1ad3408fb 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 3632cd8c1..6bab799de 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 8eca427a1..0b94c134b 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