Fix guided topic modeling in cuML's UMAP (MaartenGr#1326)

lmcinnes · Jul 11, 2023 · 68de971 · 68de971
1 parent 8e0e316
commit 68de971
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bertopic/_bertopic.py b/bertopic/_bertopic.py
@@ -3180,6 +3180,8 @@ def _reduce_dimensionality(self,
         # Regular fit
         else:
             try:
+                # cuml umap needs y to be an numpy array
+                y = np.array(y) if y is not None else None
                 self.umap_model.fit(embeddings, y=y)
             except TypeError:
                 logger.info("The dimensionality reduction algorithm did not contain the `y` parameter and"