[Fix] error in label filtering

src/dataloading.py

@@ -103,11 +103,9 @@ def filter_labels(dataset_name, doc):
     labels = []
     test_target, target_key = get_test_target(doc[0])
     if "answer_index" in doc[0].keys():
-        for d in doc:
-            labels.append(d["answer_index"])
-    elif test_target.starts_with("("):
-        for d in doc:
-            labels.append(opt_in_pars_to_index(d[target_key]))
+        labels = [d["answer_index"] for d in doc]
+    elif test_target.startswith("("):
+        labels = [opt_in_pars_to_index(d[target_key]) for d in doc]
     elif dataset_name in ["bbh_boolean_expression"]:
         for d in doc:
             if d[target_key] == "True":