Skip to content

Commit f124355

Browse files
author
elasticsearchmachine
committed
[CI] Auto commit changes from spotless
1 parent 11b94cc commit f124355

File tree

2 files changed

+11
-17
lines changed

2 files changed

+11
-17
lines changed

server/src/main/java/org/elasticsearch/search/vectors/DirectIOVectorBatchLoader.java

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -73,9 +73,10 @@ private Map<Integer, float[]> loadSegmentVectorsParallel(int[] docIds, LeafReade
7373
}
7474

7575
private Map<Integer, float[]> loadVectorBatch(
76-
FloatVectorValues vectorValues,
77-
List<Integer> docIdBatch,
78-
Map<Integer, Integer> docToOrdinal) throws IOException {
76+
FloatVectorValues vectorValues,
77+
List<Integer> docIdBatch,
78+
Map<Integer, Integer> docToOrdinal
79+
) throws IOException {
7980

8081
Map<Integer, float[]> batchResult = new HashMap<>();
8182

@@ -91,9 +92,7 @@ private Map<Integer, float[]> loadVectorBatch(
9192
return batchResult;
9293
}
9394

94-
private Map<Integer, Integer> buildDocToOrdinalMapping(
95-
FloatVectorValues vectorValues,
96-
int[] targetDocIds) throws IOException {
95+
private Map<Integer, Integer> buildDocToOrdinalMapping(FloatVectorValues vectorValues, int[] targetDocIds) throws IOException {
9796

9897
Map<Integer, Integer> docToOrdinal = new HashMap<>();
9998

server/src/test/java/org/elasticsearch/search/vectors/BulkVectorFunctionScoreQueryTests.java

Lines changed: 6 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -207,18 +207,12 @@ public void testParallelVectorLoading() throws IOException {
207207

208208
// Get initial documents
209209
TopDocs topDocs = searcher.search(new MatchAllDocsQuery(), 20);
210-
int[] docIds = Arrays.stream(topDocs.scoreDocs)
211-
.mapToInt(scoreDoc -> scoreDoc.doc)
212-
.toArray();
210+
int[] docIds = Arrays.stream(topDocs.scoreDocs).mapToInt(scoreDoc -> scoreDoc.doc).toArray();
213211

214212
// Test parallel loading
215213
DirectIOVectorBatchLoader batchLoader = new DirectIOVectorBatchLoader();
216214

217-
Map<Integer, float[]> parallelResult = batchLoader.loadSegmentVectors(
218-
docIds,
219-
reader.leaves().get(0),
220-
VECTOR_FIELD
221-
);
215+
Map<Integer, float[]> parallelResult = batchLoader.loadSegmentVectors(docIds, reader.leaves().get(0), VECTOR_FIELD);
222216

223217
// use regular vector loader
224218
Map<Integer, float[]> sequentialResult = new HashMap<>();
@@ -229,16 +223,17 @@ public void testParallelVectorLoading() throws IOException {
229223
// Verify results are identical
230224
assertThat(
231225
"Parallel and sequential results should have same size",
232-
parallelResult.size(), equalTo(sequentialResult.size()));
226+
parallelResult.size(),
227+
equalTo(sequentialResult.size())
228+
);
233229

234230
for (int docId : docIds) {
235231
float[] parallelVector = parallelResult.get(docId);
236232
float[] sequentialVector = sequentialResult.get(docId);
237233

238234
assertNotNull("Parallel result should contain vector for doc " + docId, parallelVector);
239235
assertNotNull("Sequential result should contain vector for doc " + docId, sequentialVector);
240-
assertArrayEquals("Vectors should be identical for doc " + docId,
241-
sequentialVector, parallelVector, 0.0001f);
236+
assertArrayEquals("Vectors should be identical for doc " + docId, sequentialVector, parallelVector, 0.0001f);
242237
}
243238
}
244239
}

0 commit comments

Comments
 (0)