Skip to content

Commit ab1bcd6

Browse files
committed
Don't use index() to populate rerank score array
1 parent 0fe2853 commit ab1bcd6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/TextSimilarityRankFeaturePhaseRankCoordinatorContext.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -194,8 +194,9 @@ protected InferenceAction.Request generateRequest(List<String> docFeatures) {
194194

195195
float[] extractScoresFromRankedDocs(List<RankedDocsResults.RankedDoc> rankedDocs) {
196196
float[] scores = new float[rankedDocs.size()];
197-
for (RankedDocsResults.RankedDoc rankedDoc : rankedDocs) {
198-
scores[rankedDoc.index()] = rankedDoc.relevanceScore();
197+
for (int i=0; i<rankedDocs.size(); i++) {
198+
RankedDocsResults.RankedDoc rankedDoc = rankedDocs.get(i);
199+
scores[i] = normalizeScore(rankedDoc.relevanceScore());
199200
}
200201
return scores;
201202
}

0 commit comments

Comments
 (0)