Correct scikit-learn dependency with keyword_indexing