Removed gensim from tests due to it not supporting newer version of s…

…cipy
x-tabdeveloping · Jul 1, 2024 · f9f0cb0 · f9f0cb0
1 parent 9fb42ec
commit f9f0cb0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 12 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -29,7 +29,7 @@ jobs:
         run: python3 -c "import sys; print(sys.version)"
 
       - name: Install dependencies
-        run: python3 -m pip install --upgrade topic-wizard pytest bertopic turftopic gensim
+        run: python3 -m pip install --upgrade topic-wizard pytest bertopic turftopic
 
       - name: Run tests
         run: python3 -m pytest tests/
diff --git a/tests/test_compatibility.py b/tests/test_compatibility.py
@@ -1,7 +1,4 @@
 from bertopic import BERTopic
-from gensim.corpora.dictionary import Dictionary
-from gensim.models import LdaModel
-from gensim.utils import tokenize
 from sklearn.datasets import fetch_20newsgroups
 
 import topicwizard
@@ -27,11 +24,11 @@ def test_bertopic():
     app = topicwizard.get_dash_app(topic_data)
 
 
-def test_gensim():
-    tokenized_corpus = [list(tokenize(text, lower=True)) for text in corpus]
-    dictionary = Dictionary(tokenized_corpus)
-    bow_corpus = [dictionary.doc2bow(text) for text in tokenized_corpus]
-    lda = LdaModel(bow_corpus, num_topics=10)
-    pipeline = gensim_pipeline(dictionary, model=lda)
-    topic_data = pipeline.prepare_topic_data(corpus)
-    app = topicwizard.get_dash_app(topic_data)
+# def test_gensim():
+#     tokenized_corpus = [list(tokenize(text, lower=True)) for text in corpus]
+#     dictionary = Dictionary(tokenized_corpus)
+#     bow_corpus = [dictionary.doc2bow(text) for text in tokenized_corpus]
+#     lda = LdaModel(bow_corpus, num_topics=10)
+#     pipeline = gensim_pipeline(dictionary, model=lda)
+#     topic_data = pipeline.prepare_topic_data(corpus)
+#     app = topicwizard.get_dash_app(topic_data)