From ca5894c4b3ba92d8877185e2ca2da38187b6d637 Mon Sep 17 00:00:00 2001
From: Andreas Hartel <andreas.hartel@aleph-alpha.com>
Date: Tue, 2 Jul 2024 10:52:18 +0200
Subject: [PATCH] reduce number of prompts for batch semantic embedding
 endpoint

---
 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 34b46f5..bcad027 100644
--- 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(