remove scikit learn dep