fixes

lotus-data · Dec 16, 2024 · 7c2c970 · 7c2c970
1 parent 3ebfd8e
commit 7c2c970
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()