Send at most 200 prompts in test_batch_embed_semantic

Aleph-Alpha · Jan 29, 2024 · 48de3c9 · 48de3c9
1 parent 87ce1aa
commit 48de3c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_embed.py b/tests/test_embed.py
@@ -214,7 +214,7 @@ def test_embed_semantic(sync_client: Client):
     assert result.num_tokens_prompt_total >= 1
 
 
-@pytest.mark.parametrize("num_prompts", [1, 100, 101, 200, 1000])
+@pytest.mark.parametrize("num_prompts", [1, 100, 101, 200])
 @pytest.mark.system_test
 def test_batch_embed_semantic(sync_client: Client, num_prompts: int):
     request = BatchSemanticEmbeddingRequest(