check for None first

MaartenGr · Jun 9, 2023 · 6f2f48e · 6f2f48e
1 parent 2fce29b
commit 6f2f48e
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 a 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"