from sklearn.feature_extraction.text import CountVectorizer

NoSuchKey