From f6ebba75381a8d9c27d37a58ba9347ae7012b0c3 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 20 Nov 2024 14:04:30 +0530 Subject: [PATCH] test: fix test --- tests/local_testing/test_whisper.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_whisper.py b/tests/local_testing/test_whisper.py index e2663a3ab..1d7b74087 100644 --- a/tests/local_testing/test_whisper.py +++ b/tests/local_testing/test_whisper.py @@ -51,9 +51,11 @@ from litellm import Router ), ], ) -@pytest.mark.parametrize("response_format", ["json", "vtt"]) +@pytest.mark.parametrize( + "response_format, timestamp_granularities", + [("json", None), ("vtt", None), ("verbose_json", ["word"])], +) @pytest.mark.parametrize("sync_mode", [True, False]) -@pytest.mark.parametrize("timestamp_granularities", [None, ["word"], ["segment"]]) @pytest.mark.asyncio async def test_transcription( model, api_key, api_base, response_format, sync_mode, timestamp_granularities