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}")