preprocess_winogrande.py

def doc_to_target(doc) -> int:
    answer_to_num = {"1": 0, "2": 1}
    return answer_to_num[doc["answer"]]


def doc_to_choice(doc) -> list[str]:
    idx = doc["sentence"].index("_") + 1
    return [doc["sentence"][idx:].strip()]


def doc_to_text(doc) -> list[str]:
    idx = doc["sentence"].index("_")
    options = [doc["option1"], doc["option2"]]
    return [doc["sentence"][:idx] + opt for opt in options]