From 14f92ca183b8b080992b5be03896358a1c3bc93b Mon Sep 17 00:00:00 2001 From: Leland McInnes Date: Tue, 14 Nov 2023 11:02:49 -0500 Subject: [PATCH] If a Series with row index is passed as labels and values it can mess things up -- just cast to array --- umap/plot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/umap/plot.py b/umap/plot.py index 467f7415..5d301521 100755 --- a/umap/plot.py +++ b/umap/plot.py @@ -1399,7 +1399,7 @@ def interactive( data = pd.DataFrame(_get_embedding(umap_object), columns=("x", "y")) if labels is not None: - data["label"] = labels + data["label"] = np.asarray(labels) if color_key is None: unique_labels = np.unique(labels) @@ -1423,7 +1423,7 @@ def interactive( colors = "color" elif values is not None: - data["value"] = values + data["value"] = np.asarray(values) palette = _to_hex(plt.get_cmap(cmap)(np.linspace(0, 1, 256))) colors = btr.linear_cmap( "value", palette, low=np.min(values), high=np.max(values)