Skip to content

Commit

Permalink
Bring back fix for metadata
Browse files Browse the repository at this point in the history
  • Loading branch information
kcz358 committed Sep 15, 2024
1 parent 10be8c3 commit 6442376
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lmms_eval/api/task.py
Original file line number Diff line number Diff line change
Expand Up @@ -448,7 +448,7 @@ def build_all_requests(

# TODO: we should override self.config.repeats if doing greedy gen so users don't waste time+compute
per_task_metadata = {"task": self.config["task"], "doc_id": doc_id, "repeats": self.config.repeats, "split": split}
if self.config.metadata:
if self.config.metadata and type(self.config.metadata) == dict: # TODO: temporary fix for metadata loading, ignore the list of dict type.
per_task_metadata.update(self.config.metadata)

inst = self.construct_requests(doc_id=doc_id, ctx=fewshot_ctx, metadata=per_task_metadata)
Expand Down

0 comments on commit 6442376

Please sign in to comment.