From f64ac62635c73b981c51db8db804106a77e73884 Mon Sep 17 00:00:00 2001 From: Garima Dhanania Date: Wed, 1 Oct 2025 13:46:06 -0700 Subject: [PATCH] removed diarization model test --- .../resources/test_transcriptions.py | 44 ------------------- 1 file changed, 44 deletions(-) diff --git a/tests/integration/resources/test_transcriptions.py b/tests/integration/resources/test_transcriptions.py index d8afd36..aae0377 100644 --- a/tests/integration/resources/test_transcriptions.py +++ b/tests/integration/resources/test_transcriptions.py @@ -179,50 +179,6 @@ def test_diarization_default(self, sync_together_client): response_dict = response.model_dump() validate_diarization_response(response_dict) - def test_diarization_nvidia(self, sync_together_client): - """ - Test diarization with nvidia model in verbose JSON format - """ - audio_url = "https://together-public-test-data.s3.us-west-2.amazonaws.com/audio/2-speaker-conversation.wav" - - response = sync_together_client.audio.transcriptions.create( - file=audio_url, - model="openai/whisper-large-v3", - response_format="verbose_json", - diarize=True, - diarization_model="nvidia", - ) - - assert isinstance(response, AudioTranscriptionVerboseResponse) - assert isinstance(response.text, str) - assert len(response.text) > 0 - - # Validate diarization fields - response_dict = response.model_dump() - validate_diarization_response(response_dict) - - def test_diarization_pyannote(self, sync_together_client): - """ - Test diarization with pyannote model in verbose JSON format - """ - audio_url = "https://together-public-test-data.s3.us-west-2.amazonaws.com/audio/2-speaker-conversation.wav" - - response = sync_together_client.audio.transcriptions.create( - file=audio_url, - model="openai/whisper-large-v3", - response_format="verbose_json", - diarize=True, - diarization_model="pyannote", - ) - - assert isinstance(response, AudioTranscriptionVerboseResponse) - assert isinstance(response.text, str) - assert len(response.text) > 0 - - # Validate diarization fields - response_dict = response.model_dump() - validate_diarization_response(response_dict) - def test_no_diarization(self, sync_together_client): """ Test with diarize=false should not have speaker segments