tidy

Author: vigyasharma

lucene/core/src/java/org/apache/lucene/search/ByteVectorSimilarityValuesSource.java

@@ -33,10 +33,13 @@
  */
 class ByteVectorSimilarityValuesSource extends VectorSimilarityValuesSource {
 
-  /** Creates a {@link ByteVectorSimilarityValuesSource} that scores on full precision vector values */
+  /**
+   * Creates a {@link ByteVectorSimilarityValuesSource} that scores on full precision vector values
+   */
   public static DoubleValues fullPrecisionScores(
       LeafReaderContext ctx, byte[] queryVector, String vectorField) throws IOException {
-    return new ByteVectorSimilarityValuesSource(queryVector, vectorField, true).getValues(ctx, null);
+    return new ByteVectorSimilarityValuesSource(queryVector, vectorField, true)
+        .getValues(ctx, null);
   }
 
   private final byte[] queryVector;
@@ -62,7 +65,8 @@ public ByteVectorSimilarityValuesSource(byte[] vector, String fieldName) {
    * @param useFullPrecision uses full precision raw vectors for similarity computation if true,
    *     otherwise the configured vectors reader is used, which may be quantized or full precision.
    */
-  public ByteVectorSimilarityValuesSource(byte[] vector, String fieldName, boolean useFullPrecision) {
+  public ByteVectorSimilarityValuesSource(
+      byte[] vector, String fieldName, boolean useFullPrecision) {
     super(fieldName);
     this.queryVector = vector;
     this.useFullPrecision = useFullPrecision;

lucene/core/src/java/org/apache/lucene/search/DoubleValuesSource.java

@@ -24,7 +24,6 @@
 import org.apache.lucene.index.DocValues;
 import org.apache.lucene.index.LeafReaderContext;
 import org.apache.lucene.index.NumericDocValues;
-import org.apache.lucene.index.VectorEncoding;
 import org.apache.lucene.search.comparators.DoubleComparator;
 import org.apache.lucene.util.NumericUtils;
 

lucene/core/src/java/org/apache/lucene/search/FloatVectorSimilarityValuesSource.java

@@ -33,10 +33,13 @@
  */
 class FloatVectorSimilarityValuesSource extends VectorSimilarityValuesSource {
 
-  /** Creates a {@link FloatVectorSimilarityValuesSource} that scores on full precision vector values */
+  /**
+   * Creates a {@link FloatVectorSimilarityValuesSource} that scores on full precision vector values
+   */
   public static DoubleValues fullPrecisionScores(
       LeafReaderContext ctx, float[] queryVector, String vectorField) throws IOException {
-    return new FloatVectorSimilarityValuesSource(queryVector, vectorField, true).getValues(ctx, null);
+    return new FloatVectorSimilarityValuesSource(queryVector, vectorField, true)
+        .getValues(ctx, null);
   }
 
   private final float[] queryVector;

lucene/core/src/test/org/apache/lucene/search/TestQuantizedVectorSimilarityValueSource.java

@@ -148,8 +148,10 @@ public void testFullPrecisionVectorSimilarityDVS() throws Exception {
       float[] queryVector = TestVectorUtil.randomVector(VECTOR_DIMENSION);
       try (IndexReader reader = DirectoryReader.open(dir)) {
         for (LeafReaderContext ctx : reader.leaves()) {
-          DoubleValues fpSimValues = FloatVectorSimilarityValuesSource.fullPrecisionScores(ctx, queryVector, KNN_FIELD);
-          DoubleValues quantizedSimValues = DoubleValuesSource.similarityToQueryVector(ctx, queryVector, KNN_FIELD);
+          DoubleValues fpSimValues =
+              FloatVectorSimilarityValuesSource.fullPrecisionScores(ctx, queryVector, KNN_FIELD);
+          DoubleValues quantizedSimValues =
+              DoubleValuesSource.similarityToQueryVector(ctx, queryVector, KNN_FIELD);
           // validate when segment has no vectors
           if (fpSimValues == DoubleValues.EMPTY || quantizedSimValues == DoubleValues.EMPTY) {
             assertEquals(fpSimValues, quantizedSimValues);
@@ -195,9 +197,9 @@ public void testFullPrecisionByteVectorSimilarityDVS() throws Exception {
 
       // index some 4 bit quantized vectors
       try (IndexWriter w =
-               new IndexWriter(
-                   dir,
-                   newIndexWriterConfig().setCodec(TestUtil.alwaysKnnVectorsFormat(getKnnFormat(4))))) {
+          new IndexWriter(
+              dir,
+              newIndexWriterConfig().setCodec(TestUtil.alwaysKnnVectorsFormat(getKnnFormat(4))))) {
         for (int j = 0; j < numSegments; j++) {
           for (int i = 0; i < numVectors; i++) {
             Document doc = new Document();
@@ -226,9 +228,9 @@ public void testFullPrecisionByteVectorSimilarityDVS() throws Exception {
 
       // index some 7 bit quantized vectors
       try (IndexWriter w =
-               new IndexWriter(
-                   dir,
-                   newIndexWriterConfig().setCodec(TestUtil.alwaysKnnVectorsFormat(getKnnFormat(7))))) {
+          new IndexWriter(
+              dir,
+              newIndexWriterConfig().setCodec(TestUtil.alwaysKnnVectorsFormat(getKnnFormat(7))))) {
         for (int j = 0; j < numSegments; j++) {
           for (int i = 0; i < numVectors; i++) {
             Document doc = new Document();
@@ -258,8 +260,10 @@ public void testFullPrecisionByteVectorSimilarityDVS() throws Exception {
       byte[] queryVector = TestVectorUtil.randomVectorBytes(VECTOR_DIMENSION);
       try (IndexReader reader = DirectoryReader.open(dir)) {
         for (LeafReaderContext ctx : reader.leaves()) {
-          DoubleValues fpSimValues = ByteVectorSimilarityValuesSource.fullPrecisionScores(ctx, queryVector, KNN_FIELD);
-          DoubleValues quantizedSimValues = DoubleValuesSource.similarityToQueryVector(ctx, queryVector, KNN_FIELD);
+          DoubleValues fpSimValues =
+              ByteVectorSimilarityValuesSource.fullPrecisionScores(ctx, queryVector, KNN_FIELD);
+          DoubleValues quantizedSimValues =
+              DoubleValuesSource.similarityToQueryVector(ctx, queryVector, KNN_FIELD);
           // validate when segment has no vectors
           if (fpSimValues == DoubleValues.EMPTY || quantizedSimValues == DoubleValues.EMPTY) {
             assertEquals(fpSimValues, quantizedSimValues);