[Fix] key error for binary datasets

src/dataloading.py

@@ -95,11 +95,11 @@ def filter_labels(doc):
         for d in doc:
             labels.append(d["answer_index"])
     else:
-        print("Keys:", doc[0].keys())
+        print("Target:", doc[0]["target"])
         for d in doc:
-            if d["answer"] == "False":
+            if d["target"] == "False":
                 labels.append(0)
-            elif d["answer"] == "True":
+            elif d["target"] == "True":
                 labels.append(1)
             else:
                 raise ValueError("Invalid label")