o 7t¾iîã@s"ddlZdejdejfdd„ZdS)éNÚdatasetÚreturncCsdd„}| |¡S)NcSsŠgd¢}|d|d|d|dg}| dd¡r$| d¡| |d¡| dd¡r6| d¡| |d¡|d|| |d ¡d œ}|S) N)ÚAÚBÚCÚDrrrrÚEÚFÚquestionÚanswer)Ú questionsÚchoicesÚgoal)ÚgetÚappendÚindex)ÚdocÚ answer_listr Úout_doc©rúT/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/tmlu/default/utils.pyÚ_helpers     ýzprocess_docs.._helper)Úmap)rrrrrÚ process_docss r)ÚdatasetsÚDatasetrrrrrÚs