from functools import partial def convert_choice(choice): return choice[0].lower() + choice[1:] def doc_to_text(doc, connector): # Drop the period conn = connector[doc["question"]] return doc["premise"].strip()[:-1] + f" {conn}" def doc_to_choice(doc): return [convert_choice(doc["choice1"]), convert_choice(doc["choice2"])] doc_to_text_et = partial( doc_to_text, connector={ "cause": "sest", "effect": "seetõttu", }, ) doc_to_text_ht = partial( doc_to_text, connector={ "cause": "poukisa", "effect": "donk sa", }, ) doc_to_text_it = partial( doc_to_text, connector={ "cause": "perché", "effect": "quindi", }, ) doc_to_text_id = partial( doc_to_text, connector={ "cause": "karena", "effect": "maka", }, ) doc_to_text_qu = partial( doc_to_text, connector={ "cause": "imataq", "effect": "chaymi", }, ) doc_to_text_sw = partial( doc_to_text, connector={ "cause": "kwa sababu", "effect": "kwa hiyo", }, ) doc_to_text_zh = partial( doc_to_text, connector={ "cause": "因为", "effect": "所以", }, ) doc_to_text_ta = partial( doc_to_text, connector={ "cause": "காரணமாக", "effect": "எனவே", }, ) doc_to_text_th = partial( doc_to_text, connector={ "cause": "เพราะ", "effect": "ดังนั้น", }, ) doc_to_text_tr = partial( doc_to_text, connector={ "cause": "çünkü", "effect": "bu yüzden", }, ) doc_to_text_vi = partial( doc_to_text, connector={ "cause": "bởi vì", "effect": "vì vậy", }, )