Spaces:
Running
Running
Joschka Strueber
commited on
Commit
·
3dfa66b
1
Parent(s):
64789e4
[Fix] error in dataset name, error in digit check for str
Browse files- src/dataloading.py +2 -2
src/dataloading.py
CHANGED
@@ -106,7 +106,7 @@ def filter_labels(dataset_name, doc):
|
|
106 |
labels = [d["answer_index"] for d in doc]
|
107 |
elif test_target.startswith("("):
|
108 |
labels = [opt_in_pars_to_index(d[target_key]) for d in doc]
|
109 |
-
elif dataset_name in ["
|
110 |
for d in doc:
|
111 |
if d[target_key] == "True":
|
112 |
labels.append(1)
|
@@ -130,7 +130,7 @@ def filter_labels(dataset_name, doc):
|
|
130 |
labels.append(0)
|
131 |
elif d[target_key] == "no":
|
132 |
labels.append(1)
|
133 |
-
elif test_target.
|
134 |
labels = [int(d[target_key]) for d in doc]
|
135 |
|
136 |
return labels
|
|
|
106 |
labels = [d["answer_index"] for d in doc]
|
107 |
elif test_target.startswith("("):
|
108 |
labels = [opt_in_pars_to_index(d[target_key]) for d in doc]
|
109 |
+
elif dataset_name in ["bbh_boolean_expressions"]:
|
110 |
for d in doc:
|
111 |
if d[target_key] == "True":
|
112 |
labels.append(1)
|
|
|
130 |
labels.append(0)
|
131 |
elif d[target_key] == "no":
|
132 |
labels.append(1)
|
133 |
+
elif test_target.isdigit():
|
134 |
labels = [int(d[target_key]) for d in doc]
|
135 |
|
136 |
return labels
|