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 authored and joein committed Jan 27, 2025
1 parent 663d7ae commit fee54d0
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 fee54d0

Please sign in to comment.