from datasets import load_dataset | |
# Load only the test set | |
dataset = load_dataset("INK-USC/riddle_sense", trust_remote_code=True)["validation"] | |
# Print first example from test set | |
example = dataset[0] | |
print("\n--- First Test Example ---") | |
print("Question:", example["question"]) | |
print("AnswerKey:", example["answerKey"]) | |
print("Choices:") | |
for label, text in zip(example["choices"]["label"], example["choices"]["text"]): | |
print(f"{label}. {text}") | |