Use approximation instead of subScores

iverase · iverase · commit 83436de3e379 · 2025-04-01T14:40:04.000+02:00
diff --git a/x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/inference/ltr/QueryFeatureExtractor.java b/x-pack/plugin/ml/src/main/java/org/elasticsearch/xpack/ml/inference/ltr/QueryFeatureExtractor.java
@@ -74,7 +74,7 @@ public void addFeatures(Map<String, Object> featureMap, int docId) throws IOExce
         if (approximation.docID() != docId) {
             return;
         }
-        var w = (FeatureDisiWrapper) subScorers.topList();
+        var w = (FeatureDisiWrapper) approximation.topList();
         for (; w != null; w = (FeatureDisiWrapper) w.next) {
             if (w.twoPhaseView == null || w.twoPhaseView.matches()) {
                 featureMap.put(w.featureName, w.scorable.score());

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,7 @@ public void addFeatures(Map<String, Object> featureMap, int docId) throws IOExce`
`74`	`74`	`if (approximation.docID() != docId) {`
`75`	`75`	`return;`
`76`	`76`	`}`
`77`		`- var w = (FeatureDisiWrapper) subScorers.topList();`
	`77`	`+ var w = (FeatureDisiWrapper) approximation.topList();`
`78`	`78`	`for (; w != null; w = (FeatureDisiWrapper) w.next) {`
`79`	`79`	`if (w.twoPhaseView == null \|\| w.twoPhaseView.matches()) {`
`80`	`80`	`featureMap.put(w.featureName, w.scorable.score());`