Skip to content

Commit adba6ad

Browse files
committed
fix
1 parent 2d7e4f1 commit adba6ad

File tree

2 files changed

+16
-16
lines changed

2 files changed

+16
-16
lines changed

paimon-core/src/test/java/org/apache/paimon/table/source/VectorSearchBuilderTest.java

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -107,10 +107,10 @@ public void testVectorSearchEndToEnd() throws Exception {
107107
assertThat(result.results().isEmpty()).isFalse();
108108

109109
// Read using the search result
110-
ReadBuilder readBuilder = table.newReadBuilder().withGlobalIndexResult(result);
110+
ReadBuilder readBuilder = table.newReadBuilder();
111111
List<Integer> ids = new ArrayList<>();
112-
try (RecordReader<InternalRow> reader =
113-
readBuilder.newRead().createReader(readBuilder.newScan().plan())) {
112+
TableScan.Plan plan = readBuilder.newScan().withGlobalIndexResult(result).plan();
113+
try (RecordReader<InternalRow> reader = readBuilder.newRead().createReader(plan)) {
114114
reader.forEachRemaining(row -> ids.add(row.getInt(0)));
115115
}
116116

@@ -157,10 +157,10 @@ public void testVectorSearchWithCosineMetric() throws Exception {
157157

158158
assertThat(result).isInstanceOf(ScoredGlobalIndexResult.class);
159159

160-
ReadBuilder readBuilder = table.newReadBuilder().withGlobalIndexResult(result);
160+
ReadBuilder readBuilder = table.newReadBuilder();
161+
TableScan.Plan plan = readBuilder.newScan().withGlobalIndexResult(result).plan();
161162
List<Integer> ids = new ArrayList<>();
162-
try (RecordReader<InternalRow> reader =
163-
readBuilder.newRead().createReader(readBuilder.newScan().plan())) {
163+
try (RecordReader<InternalRow> reader = readBuilder.newRead().createReader(plan)) {
164164
reader.forEachRemaining(row -> ids.add(row.getInt(0)));
165165
}
166166

@@ -209,10 +209,10 @@ public void testVectorSearchTopKLimit() throws Exception {
209209
.withVectorColumn(VECTOR_FIELD_NAME)
210210
.executeLocal();
211211

212-
ReadBuilder readBuilder = table.newReadBuilder().withGlobalIndexResult(result);
212+
ReadBuilder readBuilder = table.newReadBuilder();
213+
TableScan.Plan plan = readBuilder.newScan().withGlobalIndexResult(result).plan();
213214
List<Integer> ids = new ArrayList<>();
214-
try (RecordReader<InternalRow> reader =
215-
readBuilder.newRead().createReader(readBuilder.newScan().plan())) {
215+
try (RecordReader<InternalRow> reader = readBuilder.newRead().createReader(plan)) {
216216
reader.forEachRemaining(row -> ids.add(row.getInt(0)));
217217
}
218218

@@ -250,10 +250,10 @@ public void testVectorSearchWithMultipleIndexFiles() throws Exception {
250250
assertThat(result).isInstanceOf(ScoredGlobalIndexResult.class);
251251
assertThat(result.results().isEmpty()).isFalse();
252252

253-
ReadBuilder readBuilder = table.newReadBuilder().withGlobalIndexResult(result);
253+
ReadBuilder readBuilder = table.newReadBuilder();
254+
TableScan.Plan plan = readBuilder.newScan().withGlobalIndexResult(result).plan();
254255
List<Integer> ids = new ArrayList<>();
255-
try (RecordReader<InternalRow> reader =
256-
readBuilder.newRead().createReader(readBuilder.newScan().plan())) {
256+
try (RecordReader<InternalRow> reader = readBuilder.newRead().createReader(plan)) {
257257
reader.forEachRemaining(row -> ids.add(row.getInt(0)));
258258
}
259259

paimon-lumina/src/test/java/org/apache/paimon/lumina/index/LuminaVectorGlobalIndexScanTest.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -138,8 +138,8 @@ public void testVectorIndexScanEndToEnd() throws Exception {
138138
.withVectorColumn(vectorFieldName);
139139
GlobalIndexResult indexResult =
140140
vectorBuilder.newVectorRead().read(vectorBuilder.newVectorScan().scan());
141-
ReadBuilder readBuilder = table.newReadBuilder().withGlobalIndexResult(indexResult);
142-
TableScan scan = readBuilder.newScan();
141+
ReadBuilder readBuilder = table.newReadBuilder();
142+
TableScan scan = readBuilder.newScan().withGlobalIndexResult(indexResult);
143143
List<Integer> ids = new ArrayList<>();
144144
readBuilder
145145
.newRead()
@@ -253,8 +253,8 @@ public PositionOutputStream newOutputStream(String fileName)
253253
.withVectorColumn(vectorFieldName);
254254
GlobalIndexResult indexResult =
255255
vectorBuilder.newVectorRead().read(vectorBuilder.newVectorScan().scan());
256-
ReadBuilder readBuilder = table.newReadBuilder().withGlobalIndexResult(indexResult);
257-
TableScan scan = readBuilder.newScan();
256+
ReadBuilder readBuilder = table.newReadBuilder();
257+
TableScan scan = readBuilder.newScan().withGlobalIndexResult(indexResult);
258258
List<Integer> ids = new ArrayList<>();
259259
readBuilder
260260
.newRead()

0 commit comments

Comments
 (0)