Joschka Strueber commited on
Commit
3dfa66b
·
1 Parent(s): 64789e4

[Fix] error in dataset name, error in digit check for str

Browse files
Files changed (1) hide show
  1. 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 ["bbh_boolean_expression"]:
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.is_digit():
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