From 9e8032ef2f0312e90b1a74370149332c83474002 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tim=20S=C3=BCberkr=C3=BCb?= Date: Tue, 3 Oct 2023 16:27:51 +0200 Subject: [PATCH] Reduce number of tasks created by batch semantic embedding test --- tests/test_embed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_embed.py b/tests/test_embed.py index c310c84..11ce668 100644 --- a/tests/test_embed.py +++ b/tests/test_embed.py @@ -52,7 +52,7 @@ async def test_can_semantic_embed_with_async_client( assert len(response.embedding) == 128 -@pytest.mark.parametrize("num_prompts", [1, 100, 101, 200, 1000]) +@pytest.mark.parametrize("num_prompts", [1, 100, 101]) @pytest.mark.parametrize("batch_size", [1, 32, 100]) @pytest.mark.system_test async def test_batch_embed_semantic_with_async_client(