From a1362cf41793b34a1fc25bc20e972091c90be7ee Mon Sep 17 00:00:00 2001 From: Advaith Rao Date: Thu, 16 Nov 2023 14:46:07 -0500 Subject: [PATCH] Updated nltk dependancies in workflows --- .github/workflows/pipeline.yml | 4 ++++ tests/test_modeler.py | 3 --- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/.github/workflows/pipeline.yml b/.github/workflows/pipeline.yml index e948106..4b9030f 100644 --- 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 index f162539..dabe15f 100644 --- 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