Checking if fetched values is empty.

afoucret · Jan 12, 2024 · dd8da55 · dd8da55
1 parent ed8c980
commit dd8da55
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...ml/src/main/java/org/elasticsearch/xpack/ml/inference/ltr/FieldValueFeatureExtractor.java b/...ml/src/main/java/org/elasticsearch/xpack/ml/inference/ltr/FieldValueFeatureExtractor.java
@@ -53,7 +53,10 @@ public void setNextReader(LeafReaderContext segmentContext) {
     public void addFeatures(Map<String, Object> featureMap, int docId) throws IOException {
         Source source = sourceLookup.getSource(this.segmentContext, docId);
         for (FieldValueFetcher vf : this.valueFetcherList) {
-            featureMap.put(vf.fieldName(), vf.valueFetcher().fetchValues(source, docId, new ArrayList<>()).get(0));
+            List<Object> values = vf.valueFetcher().fetchValues(source, docId, new ArrayList<>());
+            if (values.isEmpty() == false) {
+                featureMap.put(vf.fieldName(), values.get(0));
+            }
         }
     }