fix: add type checks for diarization results in transcribe_task function

Author: Barabazs

whisperx/transcribe.py

@@ -4,6 +4,7 @@
 import warnings
 
 import numpy as np
+import pandas as pd
 import torch
 
 from whisperx.alignment import align, load_align_model
@@ -218,15 +219,23 @@ def transcribe_task(args: dict, parser: argparse.ArgumentParser):
         diarize_model = DiarizationPipeline(model_name=diarize_model_name, use_auth_token=hf_token, device=device)
         for result, input_audio_path in tmp_results:
             diarize_result = diarize_model(
-                input_audio_path, 
-                min_speakers=min_speakers, 
-                max_speakers=max_speakers, 
+                input_audio_path,
+                min_speakers=min_speakers,
+                max_speakers=max_speakers,
                 return_embeddings=return_speaker_embeddings
             )
 
             if return_speaker_embeddings:
+                if not isinstance(diarize_result, tuple):
+                    raise TypeError(
+                        f"Expected tuple when return_embeddings=True, got {type(diarize_result).__name__}"
+                    )
                 diarize_segments, speaker_embeddings = diarize_result
             else:
+                if not isinstance(diarize_result, pd.DataFrame):
+                    raise TypeError(
+                        f"Expected DataFrame when return_embeddings=False, got {type(diarize_result).__name__}"
+                    )
                 diarize_segments = diarize_result
                 speaker_embeddings = None