Updated the name of the variable data_samples_long to data_samples

README.md

@@ -160,8 +160,8 @@ def COPA_eval(prompt, alternative1, alternative2):
     lprob2 = get_logprobs(prompt + "\n" + alternative2).sum()
     return 0 if lprob1 > lprob2 else 1
 
-for lang in data_samples_long:
-    for idx, example in enumerate(data_samples_long[lang]):
+for lang in data_samples:
+    for idx, example in enumerate(data_samples[lang]):
         predict = COPA_eval(example["premise"], example["choice1"], example["choice2"])
         print(f'{lang}-{idx}', predict, example['label'])