vlad.os/whisperx-rocm-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: rename types.py to schema.py to avoid stdlib conflict

Browse Source
This commit is contained in:
Barabazs
2025-10-09 12:22:58 +00:00
parent 027ec57aee
commit 3b1b9a8c4d
5 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
whisperx/alignment.py
View File

@@ -15,7 +15,7 @@ from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
from whisperx.audio import SAMPLE_RATE, load_audio from whisperx.audio import SAMPLE_RATE, load_audio
from whisperx.utils import interpolate_nans from whisperx.utils import interpolate_nans
from whisperx.types import ( from whisperx.schema import (
AlignedTranscriptionResult, AlignedTranscriptionResult,
SingleSegment, SingleSegment,
SingleAlignedSegment, SingleAlignedSegment,

2
whisperx/asr.py
View File

@@ -12,7 +12,7 @@ from transformers import Pipeline
from transformers.pipelines.pt_utils import PipelineIterator from transformers.pipelines.pt_utils import PipelineIterator
from whisperx.audio import N_SAMPLES, SAMPLE_RATE, load_audio, log_mel_spectrogram 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 from whisperx.vads import Vad, Silero, Pyannote

2
whisperx/diarize.py
View File

@@ -5,7 +5,7 @@ from typing import Optional, Union
import torch import torch
from whisperx.audio import load_audio, SAMPLE_RATE from whisperx.audio import load_audio, SAMPLE_RATE
from whisperx.types import TranscriptionResult, AlignedTranscriptionResult from whisperx.schema import TranscriptionResult, AlignedTranscriptionResult
class DiarizationPipeline: class DiarizationPipeline:

0
whisperx/types.py → whisperx/schema.py
View File

2
whisperx/transcribe.py
View File

@@ -10,7 +10,7 @@ from whisperx.alignment import align, load_align_model
from whisperx.asr import load_model from whisperx.asr import load_model
from whisperx.audio import load_audio from whisperx.audio import load_audio
from whisperx.diarize import DiarizationPipeline, assign_word_speakers 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 from whisperx.utils import LANGUAGES, TO_LANGUAGE_CODE, get_writer