diff --git a/lotus/utils.py b/lotus/utils.py index 769fe940..f1547494 100644 --- a/lotus/utils.py +++ b/lotus/utils.py @@ -61,7 +61,7 @@ def ret( scores, indices = kmeans.index.search(vec_set, 1) # get the cluster centroids - centroids = kmeans.centroids + # centroids = kmeans.centroids # return indices.flatten(), scores.flatten(), centroids return indices.flatten()