File size: 178 Bytes
9d5b280
 
 
 
 
 
1
2
3
4
5
6
7
def process_docs(dataset):
    def _helper(doc):
        doc["choices"] = [doc["Option A"], doc["Option B"], doc["Option C"]]
        return doc

    return dataset.map(_helper)