diff --git a/bertopic/_bertopic.py b/bertopic/_bertopic.py index 753515d7..34273467 100644 --- a/bertopic/_bertopic.py +++ b/bertopic/_bertopic.py @@ -3685,11 +3685,9 @@ def _zeroshot_topic_modeling(self, documents: pd.DataFrame, embeddings: np.ndarr def _is_zeroshot(self): """ Check whether zero-shot topic modeling is possible - * There should be a cluster model used * Embedding model is necessary to convert zero-shot topics to embeddings * Zero-shot topics should be defined """ - # TODO how to handle check for type(self.hdbscan_model) != BaseCluster if self.zeroshot_topic_list is not None and self.embedding_model is not None: return True return False