Was going though @jacobtomlinson dask tutorial and I saw

from dask_ml.metrics import mean_squared_error
from math import sqrt

sqrt(mean_squared_error(y_test, y_predicted))

Wonder if it's possible to add the squared arg - same as scikit-learn (https://scikit-learn.org/stable/modules/generated/sklearn.metrics.mean_squared_error.html; https://github.com/scikit-learn/scikit-learn/blob/fd237278e/sklearn/metrics/_regression.py#L266)

May be a one line addition?