diff --git a/bertopic/_bertopic.py b/bertopic/_bertopic.py index 22b6c03c..d29505ce 100644 --- a/bertopic/_bertopic.py +++ b/bertopic/_bertopic.py @@ -456,7 +456,10 @@ def transform(self, """ After having fit a model, use transform to predict new instances Arguments: - documents: A single document or a list of documents to predict on + documents: A single document or a list of documents to predict on. + Note that the behavior of the method might differ depending + on whether a single document or a list of documents is passed + (especially when using the HDBSCAN algorithm) embeddings: Pre-trained document embeddings. These can be used instead of the sentence-transformer model. images: A list of paths to the images to predict on or the images themselves