Fix pipeline (#2008)

MaartenGr · May 23, 2024 · a2b19d1 · a2b19d1
1 parent ccc9ebd
commit a2b19d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -125,7 +125,7 @@ def supervised_topic_model(documents, document_embeddings, embedding_model, targ
 
 @pytest.fixture(scope="session")
 def online_topic_model(documents, document_embeddings, embedding_model):
-    umap_model = IncrementalPCA(n_components=5)
+    umap_model = PCA(n_components=5)
     cluster_model = MiniBatchKMeans(n_clusters=50, random_state=0)
     vectorizer_model = OnlineCountVectorizer(stop_words="english", decay=.01)
     model = BERTopic(umap_model=umap_model, hdbscan_model=cluster_model, vectorizer_model=vectorizer_model, embedding_model=embedding_model)