Add scikit-learn dependency