fix: handle pyannote 4.x DiarizeOutput wrapper in diarize()

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-04-02 11:34:02 +02:00
parent 8ee11a31a1
commit 180fe43df7
2 changed files with 6 additions and 2 deletions
@@ -20,7 +20,9 @@ class Diarizer:
    async def diarize(self, wav_path: str) -> list[tuple[float, float, str]]:
        loop = asyncio.get_running_loop()
        pipeline = await loop.run_in_executor(None, self._load_pipeline)
-        annotation = await loop.run_in_executor(None, lambda: pipeline(wav_path))
+        result = await loop.run_in_executor(None, lambda: pipeline(wav_path))
+        # pyannote 4.x returns DiarizeOutput; older versions return Annotation directly
+        annotation = getattr(result, "speaker_diarization", result)
        return [
            (turn.start, turn.end, speaker)
            for turn, _, speaker in annotation.itertracks(yield_label=True)