Skip to content

Commit 4384c74

Browse files
authored
removed diarization model test (#371)
1 parent bf000f4 commit 4384c74

File tree

1 file changed

+0
-44
lines changed

1 file changed

+0
-44
lines changed

tests/integration/resources/test_transcriptions.py

Lines changed: 0 additions & 44 deletions
Original file line numberDiff line numberDiff line change
@@ -179,50 +179,6 @@ def test_diarization_default(self, sync_together_client):
179179
response_dict = response.model_dump()
180180
validate_diarization_response(response_dict)
181181

182-
def test_diarization_nvidia(self, sync_together_client):
183-
"""
184-
Test diarization with nvidia model in verbose JSON format
185-
"""
186-
audio_url = "https://together-public-test-data.s3.us-west-2.amazonaws.com/audio/2-speaker-conversation.wav"
187-
188-
response = sync_together_client.audio.transcriptions.create(
189-
file=audio_url,
190-
model="openai/whisper-large-v3",
191-
response_format="verbose_json",
192-
diarize=True,
193-
diarization_model="nvidia",
194-
)
195-
196-
assert isinstance(response, AudioTranscriptionVerboseResponse)
197-
assert isinstance(response.text, str)
198-
assert len(response.text) > 0
199-
200-
# Validate diarization fields
201-
response_dict = response.model_dump()
202-
validate_diarization_response(response_dict)
203-
204-
def test_diarization_pyannote(self, sync_together_client):
205-
"""
206-
Test diarization with pyannote model in verbose JSON format
207-
"""
208-
audio_url = "https://together-public-test-data.s3.us-west-2.amazonaws.com/audio/2-speaker-conversation.wav"
209-
210-
response = sync_together_client.audio.transcriptions.create(
211-
file=audio_url,
212-
model="openai/whisper-large-v3",
213-
response_format="verbose_json",
214-
diarize=True,
215-
diarization_model="pyannote",
216-
)
217-
218-
assert isinstance(response, AudioTranscriptionVerboseResponse)
219-
assert isinstance(response.text, str)
220-
assert len(response.text) > 0
221-
222-
# Validate diarization fields
223-
response_dict = response.model_dump()
224-
validate_diarization_response(response_dict)
225-
226182
def test_no_diarization(self, sync_together_client):
227183
"""
228184
Test with diarize=false should not have speaker segments

0 commit comments

Comments
 (0)