Use a more recent version of sklearn