Updated nltk dependancies in workflows

advaithsrao · Nov 16, 2023 · a1362cf · a1362cf
1 parent aa9e591
commit a1362cf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/.github/workflows/pipeline.yml b/.github/workflows/pipeline.yml
@@ -43,6 +43,10 @@ jobs:
         poetry install
       working-directory: ${{ github.workspace }}
 
+    - name: NLTK dependancies
+      run: |
+        poetry run python3 -m nltk.downloader stopwords
+
     - name: Run tests
       run: |
         poetry run pytest -n 4 -v tests/

diff --git a/tests/test_modeler.py b/tests/test_modeler.py
@@ -79,9 +79,6 @@ 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