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