Update Space (evaluate main: c447fc8e)

mse.py

@@ -13,9 +13,6 @@
 # limitations under the License.
 """MSE - Mean Squared Error Metric"""
 
-from dataclasses import dataclass
-from typing import List, Optional
-
 import datasets
 from sklearn.metrics import mean_squared_error
 
@@ -88,28 +85,13 @@ Examples:
 """
 
 
-@dataclass
-class MseConfig(evaluate.info.Config):
-
-    name: str = "default"
-
-    multioutput: str = "uniform_average"
-    sample_weight: Optional[List[float]] = None
-    squared: bool = True
-
-
 @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
 class Mse(evaluate.Metric):
-
-    CONFIG_CLASS = MseConfig
-    ALLOWED_CONFIG_NAMES = ["default", "multilist"]
-
-    def _info(self, config):
+    def _info(self):
         return evaluate.MetricInfo(
             description=_DESCRIPTION,
             citation=_CITATION,
             inputs_description=_KWARGS_DESCRIPTION,
-            config=config,
             features=datasets.Features(self._get_feature_types()),
             reference_urls=[
                 "https://scikit-learn.org/stable/modules/generated/sklearn.metrics.mean_squared_error.html"
@@ -128,14 +110,10 @@ class Mse(evaluate.Metric):
                 "references": datasets.Value("float"),
             }
 
-    def _compute(self, predictions, references):
+    def _compute(self, predictions, references, sample_weight=None, multioutput="uniform_average", squared=True):
 
         mse = mean_squared_error(
-            references,
-            predictions,
-            sample_weight=self.config.sample_weight,
-            multioutput=self.config.multioutput,
-            squared=self.config.squared,
+            references, predictions, sample_weight=sample_weight, multioutput=multioutput, squared=squared
         )
 
         return {"mse": mse}

requirements.txt

@@ -1,2 +1,2 @@
-git+https://github.com/huggingface/evaluate@e4a2724377909fe2aeb4357e3971e5a569673b39
+git+https://github.com/huggingface/evaluate@c447fc8eda9c62af501bfdc6988919571050d950
 sklearn