update call to `mean_squared_error` to comply with sklearn v1.6 dbb743c verified Natooz commited on Jan 24