Skip to content

Commit

Permalink
chore: Update default task in tests
Browse files Browse the repository at this point in the history
  • Loading branch information
hh-space-invader committed Jan 20, 2025
1 parent f91ad9b commit c84ef3a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_text_multitask_embeddings.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@
def test_batch_embedding():
is_ci = os.getenv("CI")
docs_to_embed = docs * 10
default_task = Task.TEXT_MATCHING
default_task = Task.RETRIEVAL_PASSAGE

for model_desc in TextEmbedding.list_supported_models():
if not is_ci and model_desc["size_in_GB"] > 1:
Expand Down

0 comments on commit c84ef3a

Please sign in to comment.