update call to `mean_squared_error` to comply with sklearn v1.6 325581b verified Nathan commited on Jan 23, 2025