Skip to content

Commit 2e121b0

Browse files
committed
removed the hard coding
1 parent 45ca068 commit 2e121b0

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

x-pack/plugin/rank-rrf/src/main/java/org/elasticsearch/xpack/rank/rrf/RRFRetrieverBuilder.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -271,11 +271,12 @@ protected RetrieverBuilder doRewrite(QueryRewriteContext ctx) {
271271
localIndicesMetadata.values(),
272272
r -> {
273273
List<RetrieverSource> retrievers = new ArrayList<>(r.size());
274-
for (var retriever : r) {
274+
float[] weights = new float[r.size()];
275+
for (int i = 0; i < r.size(); i++) {
276+
var retriever = r.get(i);
275277
retrievers.add(retriever.retrieverSource());
278+
weights[i] = retriever.weight();
276279
}
277-
float[] weights = new float[retrievers.size()];
278-
Arrays.fill(weights, 1.0f);
279280
return new RRFRetrieverBuilder(retrievers, null, null, rankWindowSize, rankConstant, weights);
280281
},
281282
w -> {

0 commit comments

Comments
 (0)