lvwerra HF staff commited on
Commit
def22d5
1 Parent(s): 8eb8b51

Update Space (evaluate main: a45df1eb)

Browse files
Files changed (1) hide show
  1. glue.py +2 -2
glue.py CHANGED
@@ -103,7 +103,7 @@ def pearson_and_spearman(preds, labels):
103
 
104
 
105
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
106
- class Glue(evaluate.EvaluationModule):
107
  def _info(self):
108
  if self.config_name not in [
109
  "sst2",
@@ -124,7 +124,7 @@ class Glue(evaluate.EvaluationModule):
124
  '["sst2", "mnli", "mnli_mismatched", "mnli_matched", '
125
  '"cola", "stsb", "mrpc", "qqp", "qnli", "rte", "wnli", "hans"]'
126
  )
127
- return evaluate.EvaluationModuleInfo(
128
  description=_DESCRIPTION,
129
  citation=_CITATION,
130
  inputs_description=_KWARGS_DESCRIPTION,
 
103
 
104
 
105
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
106
+ class Glue(evaluate.Metric):
107
  def _info(self):
108
  if self.config_name not in [
109
  "sst2",
 
124
  '["sst2", "mnli", "mnli_mismatched", "mnli_matched", '
125
  '"cola", "stsb", "mrpc", "qqp", "qnli", "rte", "wnli", "hans"]'
126
  )
127
+ return evaluate.MetricInfo(
128
  description=_DESCRIPTION,
129
  citation=_CITATION,
130
  inputs_description=_KWARGS_DESCRIPTION,