o 6ti@s ddlZddZdZddZdS)NcCst|dS)z$ Convert a doc to a choice. choices)ast literal_eval)docrX/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/leaderboard/musr/utils.py doc_to_choicesrz*{narrative} {question} {choices} Answer:cCsTd}tt|dD]\}}||dd|d7}q tj|d|d|d}|S) z Convert a doc to text. rz -  narrativequestion)r r r) enumeraterr DOC_TO_TEXTformat)rrichoicetextrrr doc_to_textsr)rrrrrrrrs