Remove sklearn requirement