From a4e7a317a72ed492e36c1bb2cc813f05663d1fff Mon Sep 17 00:00:00 2001
From: Advaith Rao <advaith652@gmail.com>
Date: Thu, 16 Nov 2023 02:26:49 -0500
Subject: [PATCH] Test fix for nltk stopwords - word embedding

---
 tests/test_modeler.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/test_modeler.py b/tests/test_modeler.py
index dabe15f..f162539 100644
--- 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