Merge pull request #142 from Aleph-Alpha/tame-embeddings-test

Reduce number of tasks created by batch semantic embedding test
Aleph-Alpha · Oct 4, 2023 · 110aa56 · 110aa56
2 parents 23ade75 + 9e8032e
commit 110aa56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(