update for current sklearn version