fix: Use k locations only for basic BF test location (#35380)

Related to #35379 Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
2024-08-09 07:52:22 +08:00 · 2024-08-09 07:52:22 +08:00 · c0ee25afd8
parent 143502e708
commit c0ee25afd8
1 changed files with 1 additions and 1 deletions
--- a/internal/storage/pk_statistics.go
+++ b/internal/storage/pk_statistics.go
@ -228,7 +228,7 @@ func (lc *BatchLocationsCache) Locations(k uint, bfType bloomfilter.BFType) [][]
 			})
 		}

-		return lc.basicLocations
+		return lo.Map(lc.basicLocations, func(locations []uint64, _ int) []uint64 { return locations[:k] })
 	case bloomfilter.BlockedBF:
 		// for block bf, we only need cache the hash result, which is a uint and only compute once for any k value
 		if len(lc.blockLocations) != len(lc.pks) {