Remove warning about scikit-learn