phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-20 14:04:30 +05:30
parent 4ddbf1395f
commit f6ebba7538
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/local_testing/test_whisper.py
View file

@ -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