lvwerra HF staff commited on
Commit
e12cd5e
1 Parent(s): 4b6d950

Update Space (evaluate main: a45df1eb)

Browse files
Files changed (1) hide show
  1. rl_reliability.py +2 -2
rl_reliability.py CHANGED
@@ -82,14 +82,14 @@ Examples:
82
 
83
 
84
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
85
- class RLReliability(evaluate.EvaluationModule):
86
  """Computes the RL Reliability Metrics."""
87
 
88
  def _info(self):
89
  if self.config_name not in ["online", "offline"]:
90
  raise KeyError("""You should supply a configuration name selected in '["online", "offline"]'""")
91
 
92
- return evaluate.EvaluationModuleInfo(
93
  module_type="metric",
94
  description=_DESCRIPTION,
95
  citation=_CITATION,
 
82
 
83
 
84
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
85
+ class RLReliability(evaluate.Metric):
86
  """Computes the RL Reliability Metrics."""
87
 
88
  def _info(self):
89
  if self.config_name not in ["online", "offline"]:
90
  raise KeyError("""You should supply a configuration name selected in '["online", "offline"]'""")
91
 
92
+ return evaluate.MetricInfo(
93
  module_type="metric",
94
  description=_DESCRIPTION,
95
  citation=_CITATION,