lvwerra HF staff commited on
Commit
99e89c5
·
1 Parent(s): ac14f48

Update Space (evaluate main: a45df1eb)

Browse files
Files changed (1) hide show
  1. super_glue.py +2 -2
super_glue.py CHANGED
@@ -145,7 +145,7 @@ def evaluate_multirc(ids_preds, labels):
145
 
146
 
147
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
148
- class SuperGlue(evaluate.EvaluationModule):
149
  def _info(self):
150
  if self.config_name not in [
151
  "boolq",
@@ -164,7 +164,7 @@ class SuperGlue(evaluate.EvaluationModule):
164
  "You should supply a configuration name selected in "
165
  '["boolq", "cb", "copa", "multirc", "record", "rte", "wic", "wsc", "wsc.fixed", "axb", "axg",]'
166
  )
167
- return evaluate.EvaluationModuleInfo(
168
  description=_DESCRIPTION,
169
  citation=_CITATION,
170
  inputs_description=_KWARGS_DESCRIPTION,
 
145
 
146
 
147
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
148
+ class SuperGlue(evaluate.Metric):
149
  def _info(self):
150
  if self.config_name not in [
151
  "boolq",
 
164
  "You should supply a configuration name selected in "
165
  '["boolq", "cb", "copa", "multirc", "record", "rte", "wic", "wsc", "wsc.fixed", "axb", "axg",]'
166
  )
167
+ return evaluate.MetricInfo(
168
  description=_DESCRIPTION,
169
  citation=_CITATION,
170
  inputs_description=_KWARGS_DESCRIPTION,