reduce number of prompts for batch semantic embedding endpoint

Aleph-Alpha · Jul 2, 2024 · ca5894c · ca5894c
1 parent ee0ec3f
commit ca5894c
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(