From 74bbc4f4608b3f5f2ca96e093eafa25b8f83cc15 Mon Sep 17 00:00:00 2001 From: Maarten Grootendorst Date: Wed, 21 Aug 2024 16:33:01 +0200 Subject: [PATCH] Fix #2100 (#2101) --- bertopic/_bertopic.py | 8 ++++---- tests/conftest.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/bertopic/_bertopic.py b/bertopic/_bertopic.py index 171136df..ea11fefc 100644 --- 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 index 3d8d49db..fd278b0f 100644 --- 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