import ast def doc_to_choice(doc): """ Convert a doc to a choice. """ return ast.literal_eval(doc["choices"]) DOC_TO_TEXT = "{narrative}\n\n{question}\n\n{choices}\nAnswer:" def doc_to_text(doc): """ Convert a doc to text. """ choices = "" for i, choice in enumerate(ast.literal_eval(doc["choices"])): choices += f"{i + 1} - {choice}\n" text = DOC_TO_TEXT.format( narrative=doc["narrative"], question=doc["question"], choices=choices ) return text