Skip to content

Commit b418f87

Browse files
committed
Fix more test compilation failures
1 parent 9119d26 commit b418f87

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

server/src/internalClusterTest/java/org/elasticsearch/search/rank/FieldBasedRerankerIT.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ public RankShardResult buildRankFeatureShardResult(SearchHits hits, int shardId)
193193
RankFeatureDoc[] rankFeatureDocs = new RankFeatureDoc[hits.getHits().length];
194194
for (int i = 0; i < hits.getHits().length; i++) {
195195
rankFeatureDocs[i] = new RankFeatureDoc(hits.getHits()[i].docId(), hits.getHits()[i].getScore(), shardId);
196-
rankFeatureDocs[i].featureData(hits.getHits()[i].field(field).getValue().toString());
196+
rankFeatureDocs[i].featureData(List.of(hits.getHits()[i].field(field).getValue().toString()));
197197
}
198198
return new RankFeatureShardResult(rankFeatureDocs);
199199
} catch (Exception ex) {
@@ -210,7 +210,7 @@ public RankFeaturePhaseRankCoordinatorContext buildRankFeaturePhaseCoordinatorCo
210210
protected void computeScores(RankFeatureDoc[] featureDocs, ActionListener<float[]> scoreListener) {
211211
float[] scores = new float[featureDocs.length];
212212
for (int i = 0; i < featureDocs.length; i++) {
213-
scores[i] = Float.parseFloat(featureDocs[i].featureData);
213+
scores[i] = Float.parseFloat(featureDocs[i].featureData.get(0));
214214
}
215215
scoreListener.onResponse(scores);
216216
}

server/src/internalClusterTest/java/org/elasticsearch/search/rank/MockedRequestActionBasedRerankerIT.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -275,7 +275,7 @@ protected void computeScores(RankFeatureDoc[] featureDocs, ActionListener<float[
275275
l.onResponse(scores);
276276
});
277277

278-
List<String> featureData = Arrays.stream(featureDocs).map(x -> x.featureData).toList();
278+
List<String> featureData = Arrays.stream(featureDocs).map(x -> x.featureData).flatMap(List::stream).toList();
279279
TestRerankingActionRequest request = generateRequest(featureData);
280280
try {
281281
ActionType<TestRerankingActionResponse> action = actionType();

0 commit comments

Comments
 (0)