Skip to content

Commit dd8f079

Browse files
committed
Remove forbidden API
1 parent bbc7081 commit dd8f079

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

lucene/core/src/java/org/apache/lucene/search/TwoPhaseKnnVectorQuery.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
import org.apache.lucene.index.FieldInfo;
2323
import org.apache.lucene.index.FloatVectorValues;
2424
import org.apache.lucene.index.LeafReaderContext;
25+
import org.apache.lucene.util.ArrayUtil;
2526

2627
public class TwoPhaseKnnVectorQuery extends KnnFloatVectorQuery {
2728

@@ -75,7 +76,9 @@ protected TopDocs getLeafResults(
7576
Arrays.sort(results.scoreDocs, (a, b) -> Float.compare(b.score, a.score));
7677

7778
// Select the top-k ScoreDocs after re-ranking
78-
ScoreDoc[] topKDocs = Arrays.copyOfRange(results.scoreDocs, 0, originalK);
79+
ScoreDoc[] topKDocs = ArrayUtil.copyOfSubArray(results.scoreDocs, 0, originalK);
80+
81+
assert topKDocs.length == originalK;
7982

8083
return new TopDocs(results.totalHits, topKDocs);
8184
}

0 commit comments

Comments
 (0)