Fix bug in mock inference service

carlosdelest · May 9, 2024 · 3684c2a · 3684c2a
1 parent 8fa22f1
commit 3684c2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.../main/java/org/elasticsearch/xpack/inference/mock/TestDenseInferenceServiceExtension.java b/.../main/java/org/elasticsearch/xpack/inference/mock/TestDenseInferenceServiceExtension.java
@@ -135,7 +135,7 @@ private List<ChunkedInferenceServiceResults> makeChunkedResults(List<String> inp
             var results = new ArrayList<ChunkedInferenceServiceResults>();
             for (int i = 0; i < input.size(); i++) {
                 double[] values = new double[dimensions];
-                for (int j = 0; j < 5; j++) {
+                for (int j = 0; j < dimensions; j++) {
                     values[j] = stringWeight(input.get(i), j);
                 }
                 results.add(