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

refactor: update type hints in diarization module (PEP 585)

Browse Source
This commit is contained in:
Radu-Sebastian Amarie
2025-04-03 13:52:55 +00:00
committed by Barabazs
parent 1631c3040f
commit 220fec9aea
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
whisperx/diarize.py
View File

@ -1,7 +1,7 @@
import numpy as np import numpy as np
import pandas as pd import pandas as pd
from pyannote.audio import Pipeline from pyannote.audio import Pipeline
from typing import Optional, Union, Tuple, Dict, List, Any 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
@ -27,7 +27,7 @@ class DiarizationPipeline:
min_speakers: Optional[int] = None, min_speakers: Optional[int] = None,
max_speakers: Optional[int] = None, max_speakers: Optional[int] = None,
return_embeddings: bool = False, return_embeddings: bool = False,
) -> Union[Tuple[pd.DataFrame, Optional[Dict[str, List[float]]]], pd.DataFrame]: ) -> Union[tuple[pd.DataFrame, Optional[dict[str, list[float]]]], pd.DataFrame]:
""" """
Perform speaker diarization on audio. Perform speaker diarization on audio.
@ -86,7 +86,7 @@ class DiarizationPipeline:
def assign_word_speakers( def assign_word_speakers(
diarize_df: pd.DataFrame, diarize_df: pd.DataFrame,
transcript_result: Union[AlignedTranscriptionResult, TranscriptionResult], transcript_result: Union[AlignedTranscriptionResult, TranscriptionResult],
speaker_embeddings: Optional[Dict[str, List[float]]] = None, speaker_embeddings: Optional[dict[str, list[float]]] = None,
fill_nearest: bool = False, fill_nearest: bool = False,
) -> Union[AlignedTranscriptionResult, TranscriptionResult]: ) -> Union[AlignedTranscriptionResult, TranscriptionResult]:
""" """