Fix #2100 (#2101)

MaartenGr · Aug 21, 2024 · 74bbc4f · 74bbc4f
1 parent 510c15e
commit 74bbc4f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/bertopic/_bertopic.py b/bertopic/_bertopic.py
@@ -4495,11 +4495,11 @@ def _auto_reduce_topics(self, documents: pd.DataFrame, use_ctfidf: bool = False)
         for key, val in sorted(mapped_topics.items()):
             mappings[val].append(key)
         mappings = {
-            topic_from: {
-                "topics_to": topics_to,
-                "topic_sizes": [self.topic_sizes_[topic] for topic in topics_to],
+            topic_to: {
+                "topics_from": topics_from,
+                "topic_sizes": [self.topic_sizes_[topic] for topic in topics_from],
             }
-            for topic_from, topics_to in mappings.items()
+            for topic_to, topics_from in mappings.items()
         }
 
         # Update documents and topics

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -114,7 +114,7 @@ def representation_topic_model(documents, document_embeddings, embedding_model):
 @pytest.fixture(scope="session")
 def reduced_topic_model(custom_topic_model, documents):
     model = copy.deepcopy(custom_topic_model)
-    model.reduce_topics(documents, nr_topics=12)
+    model.reduce_topics(documents, nr_topics="auto")
     return model