From 588e1f724c9f5de528eb1500b0c85a1a609fe947 Mon Sep 17 00:00:00 2001 From: Leland McInnes Date: Wed, 18 Jan 2023 11:20:49 -0500 Subject: [PATCH] cast rows appropriately --- umap/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/umap/utils.py b/umap/utils.py index 41332dee..6bdffc21 100644 --- a/umap/utils.py +++ b/umap/utils.py @@ -156,7 +156,7 @@ def csr_unique(matrix, return_index=True, return_inverse=True, return_counts=Tru unique_matrix[inverse] """ lil_matrix = matrix.tolil() - rows = [x + y for x, y in zip(lil_matrix.rows, lil_matrix.data)] + rows = np.asarray([tuple(x + y) for x, y in zip(lil_matrix.rows, lil_matrix.data)], dtype=object) return_values = return_counts + return_inverse + return_index return np.unique( rows,