Skip to content

Commit e67062c

Browse files
author
Yada Pruksachatkun
authored
Fix get labels (#834)
* black style * fixing label loading * multilabel -> label * fixing * fixing nit * getting rid of one hot vectorization * fixing all_labels function callk * delete get_all_labels
1 parent c787a52 commit e67062c

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

jiant/tasks/tasks.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2263,7 +2263,6 @@ def __init__(
22632263
self._iters_by_split = None
22642264

22652265
self.label_file = os.path.join(path, label_file)
2266-
self.all_labels = None
22672266
self.n_classes = None
22682267
self._label_namespace = self.name + "_labels"
22692268

@@ -2350,9 +2349,6 @@ def _map_fn(r, idx):
23502349

23512350
return map(_map_fn, records, itertools.count())
23522351

2353-
def get_all_labels(self) -> List[str]:
2354-
return self.all_labels
2355-
23562352
def get_sentences(self) -> Iterable[Sequence[str]]:
23572353
""" Yield sentences, used to compute vocabulary. """
23582354
for split, iter in self._iters_by_split.items():

0 commit comments

Comments
 (0)