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