update call to `mean_squared_error` to comply with sklearn v1.6

mse.py

@@ -112,8 +112,13 @@ 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
-        )
+        if squared:
+            mse = mean_squared_error(
+                references, predictions, sample_weight=sample_weight, multioutput=multioutput
+            )
+        else:
+            mse = root_mean_squared_error(
+                references, predictions, sample_weight=sample_weight, multioutput=multioutput
+            )
 
         return {"mse": mse}