Skip to content

Commit

Permalink
Merge pull request #175 from Aleph-Alpha/reduce-number-of-prompts-for…
Browse files Browse the repository at this point in the history
…-batch-semantic-embedding-test

reduce number of prompts for batch semantic embedding endpoint
  • Loading branch information
ahartel authored Jul 2, 2024
2 parents ee0ec3f + ca5894c commit 18a772f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_embed.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 18a772f

Please sign in to comment.