o 6ti@s8ddlmZgdZdefddZdeefddZdS) )List)ABCDreturncsh|dt}|dkrtdtd|}dfddt|D}d|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 S/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/eus_reading/utils.py s"z'doc_to_text_context..z Pasartea: contextz Galdera: questionz Erantzuna:)len ValueErrorlettersjoin enumerate)doc num_choiceschoicesformatted_choicesr rrdoc_to_text_contexts   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