Fix calls to self.verbose in class methods (#2039)

MaartenGr · Jun 13, 2024 · b544ba7 · b544ba7
1 parent df7116d
commit b544ba7
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/bertopic/_bertopic.py b/bertopic/_bertopic.py
@@ -3162,7 +3162,7 @@ def load(cls,
             with open(file_or_dir, 'rb') as file:
                 if embedding_model:
                     topic_model = joblib.load(file)
-                    topic_model.embedding_model = select_backend(embedding_model, verbose=self.verbose)
+                    topic_model.embedding_model = select_backend(embedding_model, verbose=topic_model.verbose)
                 else:
                     topic_model = joblib.load(file)
                 return topic_model
@@ -3179,7 +3179,7 @@ def load(cls,
 
         # Replace embedding model if one is specifically chosen
         if embedding_model is not None:
-            topic_model.embedding_model = select_backend(embedding_model)
+            topic_model.embedding_model = select_backend(embedding_model, verbose=topic_model.verbose)
 
         return topic_model
 
@@ -3312,8 +3312,9 @@ def merge_models(cls, models, min_similarity: float = .7, embedding_model=None):
         merged_model.embedding_model = models[0].embedding_model
 
         # Replace embedding model if one is specifically chosen
+        verbose = any([model.verbose for model in models])
         if embedding_model is not None and type(merged_model.embedding_model) == BaseEmbedder:
-            merged_model.embedding_model = select_backend(embedding_model, verbose=self.verbose)
+            merged_model.embedding_model = select_backend(embedding_model, verbose=verbose)
         return merged_model
 
     def push_to_hf_hub(