o 6tiY@s8ddlmZgdZdefddZdeefddZdS) )List)ABCDreturncs^|dt}|dkrtdtd|}dfddt|D}d|d d|d S) z Converts a document to a formatted string. Args: doc (dict): A dictionary containing the document information. Returns: str: A formatted string containing the question and answer choices. candidatesInvalid number of candidatesN cs"g|] \}}|d|qS)z: ).0ichoicerr R/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/eus_trivia/utils.py s"zdoc_to_text..z Galdera: questionz Erantzuna:)len ValueErrorlettersjoin enumerate)doc num_choiceschoicesformatted_choicesr rr doc_to_texts  rcCs(t|d}|dkrtdtd|S)z Returns the answer choices for a document. Args: doc (dict): A dictionary containing the document information. Returns: list: A list of strings containing the answer choices. rr r N)rrr)rrr r r doc_to_choices  rN)typingrrstrrrr r r rs