Skip to content

Commit fa60856

Browse files
committed
Adjusting PR #15415 for backport
1 parent eb01732 commit fa60856

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

lucene/core/src/test/org/apache/lucene/codecs/lucene104/TestLucene104ScalarQuantizedVectorsFormat.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
import org.apache.lucene.codecs.KnnVectorsReader;
3434
import org.apache.lucene.codecs.lucene104.Lucene104ScalarQuantizedVectorsFormat.ScalarEncoding;
3535
import org.apache.lucene.codecs.lucene95.OrdToDocDISIReaderConfiguration;
36+
import org.apache.lucene.codecs.perfield.PerFieldKnnVectorsFormat;
3637
import org.apache.lucene.document.Document;
3738
import org.apache.lucene.document.KnnFloatVectorField;
3839
import org.apache.lucene.index.CodecReader;
@@ -271,7 +272,9 @@ public void testReadQuantizedVectorWithEmptyRawVectors() throws Exception {
271272
LeafReader r = getOnlyLeafReader(reader);
272273
if (r instanceof CodecReader codecReader) {
273274
KnnVectorsReader knnVectorsReader = codecReader.getVectorReader();
274-
knnVectorsReader = knnVectorsReader.unwrapReaderForField(vectorFieldName);
275+
if (knnVectorsReader instanceof PerFieldKnnVectorsFormat.FieldsReader fieldsReader) {
276+
knnVectorsReader = fieldsReader.getFieldReader(vectorFieldName);
277+
}
275278
if (knnVectorsReader instanceof Lucene104ScalarQuantizedVectorsReader quantizedReader) {
276279
FloatVectorValues floatVectorValues =
277280
quantizedReader.getFloatVectorValues(vectorFieldName);

0 commit comments

Comments
 (0)