fix: [2.3] Use k locations only for basic BF test location (milvus-io…

…#35390) Cherry-pick from master pr: milvus-io#35380 Related to milvus-io#35379 Signed-off-by: Congqi Xia <[email protected]>
weiliu1031 · Aug 9, 2024 · f75b886 · f75b886
1 parent e5681e5
commit f75b886
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/internal/storage/pk_statistics.go b/internal/storage/pk_statistics.go
@@ -208,7 +208,7 @@ func (lc *BatchLocationsCache) Locations(k uint) [][]uint64 {
 		})
 	}
 
-	return lc.locations
+	return lo.Map(lc.locations, func(locations []uint64, _ int) []uint64 { return locations[:k] })
 }
 
 func NewBatchLocationsCache(pks []PrimaryKey) *BatchLocationsCache {