Erro squared=False - Aula DummyRegressor

Boa tarde, para que estiver tendo o erro do squared = False no comando da aula, abaixo esta o código atualizado para a versão nova do Sklearn:

"Na versão mais recente do scikit-learn, o parâmetro squared foi removido da função meansquarederror. O padrão agora é retornar o MSE (Mean Squared Error). Para obter o RMSE, você precisa calcular a raiz quadrada do MSE."

from sklearn.metrics import mean_squared_error, mean_absolute_error, r2_score
import numpy as np

def calcular_metricas_regressao(y_test, y_pred):
    mse = mean_squared_error(y_test, y_pred)
    rmse = np.sqrt(mse)  # Calcula a raiz quadrada do MSE
    mae = mean_absolute_error(y_test, y_pred)
    r2 = r2_score(y_test, y_pred)
    metricas = {
        'Raiz do Erro Quadrático Médio': round(rmse, 4),
        'Erro Absoluto Médio': round(mae, 4),
        'R2 Score': round(r2, 4)
    }
    return metricas