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
@@ -21,7 +21,9 @@ def test_diarizer_returns_list_of_tuples(tmp_path):
        (mock_turn_2, "B", "SPEAKER_01"),
    ]

-    mock_pipeline = MagicMock(return_value=mock_annotation)
+    mock_output = MagicMock()
+    mock_output.speaker_diarization = mock_annotation
+    mock_pipeline = MagicMock(return_value=mock_output)

    import asyncio
    from diarization import Diarizer