Test fix for nltk stopwords - word embedding

advaithsrao · Nov 16, 2023 · a4e7a31 · a4e7a31
1 parent 2494013
commit a4e7a31
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_modeler.py b/tests/test_modeler.py
@@ -79,6 +79,9 @@ def test_evaluate_and_log(x, y_true, y_pred):
     assert os.path.exists('/tmp/test.log')
 
 def test_word2vec_embedding(mail):
+    import nltk
+    nltk.download('stopwords')
+
     embedder = Word2VecEmbedder()
     embedding = embedder.transform(mail)[0]
     assert len(embedding) == 300