Update Space (evaluate main: 8fd64d2f)

mse.py

@@ -14,7 +14,7 @@
 """MSE - Mean Squared Error Metric"""
 
 import datasets
-from sklearn.metrics import mean_squared_error
+from sklearn.metrics import mean_squared_error, root_mean_squared_error
 
 import evaluate
 
@@ -112,8 +112,10 @@ class Mse(evaluate.Metric):
 
     def _compute(self, predictions, references, sample_weight=None, multioutput="uniform_average", squared=True):
 
-        mse = mean_squared_error(
-            references, predictions, sample_weight=sample_weight, multioutput=multioutput, squared=squared
+        mse = (
+            mean_squared_error(references, predictions, sample_weight=sample_weight, multioutput=multioutput)
+            if squared
+            else root_mean_squared_error(references, predictions, sample_weight=sample_weight, multioutput=multioutput)
         )
 
         return {"mse": mse}

requirements.txt

@@ -1,2 +1,2 @@
-git+https://github.com/huggingface/evaluate@eae4d42cc150c3a61b39c6aa3860df87cdbdf77d
+git+https://github.com/huggingface/evaluate@8fd64d2f5193b7e1da945abf04da26c1ff3d8a54
 scikit-learn