We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2fce29b commit 6f2f48eCopy full SHA for 6f2f48e
bertopic/_bertopic.py
@@ -3180,6 +3180,8 @@ def _reduce_dimensionality(self,
3180
# Regular fit
3181
else:
3182
try:
3183
+ # cuml umap needs y to be a numpy array
3184
+ y = np.array(y) if y is not None else None
3185
self.umap_model.fit(embeddings, y=y)
3186
except TypeError:
3187
logger.info("The dimensionality reduction algorithm did not contain the `y` parameter and"
0 commit comments