Merge pull request #254 from voxel51/bugfix/compute-uniqueness-similarity-index

afoley587 · web-flow · commit 00ec9c6d344a · 2025-05-05T16:42:43.000-04:00
Fix `compute_uniqueness()` with custom `similarity_index`
diff --git a/fiftyone/brain/internal/core/uniqueness.py b/fiftyone/brain/internal/core/uniqueness.py
@@ -171,7 +171,7 @@ def _compute_uniqueness(
         dists = []
         with fou.ProgressBar(total=num_embeddings, progress=progress) as pb:
             for _embeddings in fou.iter_slices(embeddings, batch_size):
-                _, _dists = similarity_index._kneighbors(
+                _, _, _dists = similarity_index._kneighbors(
                     query=_embeddings, k=K + 1, return_dists=True
                 )
                 dists.extend(_dists)

Original file line number	Diff line number	Diff line change
`@@ -171,7 +171,7 @@ def _compute_uniqueness(`
`171`	`171`	`dists = []`
`172`	`172`	`with fou.ProgressBar(total=num_embeddings, progress=progress) as pb:`
`173`	`173`	`for _embeddings in fou.iter_slices(embeddings, batch_size):`
`174`		`- _, _dists = similarity_index._kneighbors(`
	`174`	`+ _, _, _dists = similarity_index._kneighbors(`
`175`	`175`	`query=_embeddings, k=K + 1, return_dists=True`
`176`	`176`	`)`
`177`	`177`	`dists.extend(_dists)`