lvwerra HF Staff commited on
Commit
5b9b37a
·
1 Parent(s): 1d73d45

Update Space (evaluate main: a45df1eb)

Browse files
Files changed (1) hide show
  1. xtreme_s.py +2 -2
xtreme_s.py CHANGED
@@ -219,14 +219,14 @@ def wer_and_cer(preds, labels, concatenate_texts, config_name):
219
 
220
 
221
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
222
- class XtremeS(evaluate.EvaluationModule):
223
  def _info(self):
224
  if self.config_name not in _CONFIG_NAMES:
225
  raise KeyError(f"You should supply a configuration name selected in {_CONFIG_NAMES}")
226
 
227
  pred_type = "int64" if self.config_name in ["fleurs-lang_id", "minds14"] else "string"
228
 
229
- return evaluate.EvaluationModuleInfo(
230
  description=_DESCRIPTION,
231
  citation=_CITATION,
232
  inputs_description=_KWARGS_DESCRIPTION,
 
219
 
220
 
221
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
222
+ class XtremeS(evaluate.Metric):
223
  def _info(self):
224
  if self.config_name not in _CONFIG_NAMES:
225
  raise KeyError(f"You should supply a configuration name selected in {_CONFIG_NAMES}")
226
 
227
  pred_type = "int64" if self.config_name in ["fleurs-lang_id", "minds14"] else "string"
228
 
229
+ return evaluate.MetricInfo(
230
  description=_DESCRIPTION,
231
  citation=_CITATION,
232
  inputs_description=_KWARGS_DESCRIPTION,