Skip to content

Commit

Permalink
Update mteb
Browse files Browse the repository at this point in the history
  • Loading branch information
Muennighoff authored Apr 8, 2024
1 parent 8a70a0f commit a342844
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.description["name"], t.description["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 a342844

Please sign in to comment.