o 6t¾i ã@s6ddlZddlZddlmZdejdejfdd„ZdS)éN)ÚDatasetÚdatasetÚreturncCsâdd„|Dƒ}g}|D]Y}d}|d|d}|dt|dƒd}|d t|d ƒd}|d t|d ƒd}|d t|dƒd}|dt|dƒd}|gd¢d|ddœ}| |¡q t |¡}t |¡}|S)NcSsg|]}|‘qS©r)Ú.0ÚsamplerrúS/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/groundcocoa/utils.pyÚ sz process_docs..a[A user has specified certain criteria for booking a flight. Below are five different flight options labeled 'A', 'B', 'C', 'D', and 'E'. Review these options and select the one that best matches the user requirements. Respond with a single option and the phrase 'The answer is Option ' followed by the correct letter - 'A', 'B', 'C', 'D', or 'E' zUser Criteria: Úqueryz Option A: zOption AÚ z Option B: zOption Bz Option C: zOption Cz Option D: zOption Dz Option E: zOption E)zThe answer is Option AzThe answer is Option BzThe answer is Option CzThe answer is Option DzThe answer is Option EzThe answer is Option ÚAnswer)ÚcriteriaÚchoicesÚgold)ÚstrÚappendÚpdÚ DataFramerÚ from_pandas)rÚ cocoa_datasetÚ processedÚdocÚquestionÚout_docÚdfrrrÚ process_docss$ ÷  r)ÚdatasetsÚpandasrrrrrrrÚs