Skip to content

Commit

Permalink
Merge pull request ContextualAI#52 from himanshushukla12/main
Browse files Browse the repository at this point in the history
Update eval_mteb.py
  • Loading branch information
Muennighoff authored Sep 3, 2024
2 parents 8a82be9 + 6629680 commit a73526e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion evaluation/eval_mteb.py
Original file line number Diff line number Diff line change
Expand Up @@ -1159,7 +1159,7 @@ def get_args():
kwargs["tasks"] = args.task_names.split(",")
elif args.task_types:
kwargs["task_types"] = args.task_types.split(",")
tasks = [(t.metadata.name, t.metadata.type) for t in MTEB(**kwargs).tasks
tasks = [(t.metadata.name, t.metadata.type) for t in MTEB(**kwargs).tasks]

if args.max_length is not None:
model.encode = partial(model.encode, max_length=args.max_length)
Expand Down

0 comments on commit a73526e

Please sign in to comment.