Fix test

carlosdelest · Apr 16, 2024 · faef7e2 · faef7e2
1 parent 7c42501
commit faef7e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...avaRestTest/java/org/elasticsearch/xpack/ml/integration/CoordinatedInferenceIngestIT.java b/...avaRestTest/java/org/elasticsearch/xpack/ml/integration/CoordinatedInferenceIngestIT.java
@@ -59,10 +59,10 @@ public void testIngestWithMultipleModelTypes() throws IOException {
             assertThat(simulatedDocs, hasSize(2));
             assertEquals(inferenceServiceModelId, MapHelper.dig("doc._source.ml.model_id", simulatedDocs.get(0)));
             var sparseEmbedding = (Map<String, Double>) MapHelper.dig("doc._source.ml.body", simulatedDocs.get(0));
-            assertEquals(Double.valueOf(2.0), sparseEmbedding.get("feature_1"));
+            assertEquals(Double.valueOf(1.0), sparseEmbedding.get("1"));
             assertEquals(inferenceServiceModelId, MapHelper.dig("doc._source.ml.model_id", simulatedDocs.get(1)));
             sparseEmbedding = (Map<String, Double>) MapHelper.dig("doc._source.ml.body", simulatedDocs.get(1));
-            assertEquals(Double.valueOf(2.0), sparseEmbedding.get("feature_1"));
+            assertEquals(Double.valueOf(1.0), sparseEmbedding.get("1"));
         }
 
         {