Revert correcting labels

MaartenGr · Aug 15, 2024 · fed0682 · fed0682
1 parent aa41c68
commit fed0682
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bertopic/_bertopic.py b/bertopic/_bertopic.py
@@ -313,7 +313,7 @@ def topic_labels_(self):
         if self._is_zeroshot():
             # Need to correct labels from zero-shot topics
             topic_id_to_zeroshot_label = {
-                self.topic_mapper_.get_mappings()[topic_id]: self.zeroshot_topic_list[zeroshot_topic_idx]
+                topic_id: self.zeroshot_topic_list[zeroshot_topic_idx]
                 for topic_id, zeroshot_topic_idx in self._topic_id_to_zeroshot_topic_idx.items()
             }
             topic_labels.update(topic_id_to_zeroshot_label)