Some Falky Test

Signed-off-by: Vikasht34 <[email protected]>
Vikasht34 · Feb 13, 2025 · b1afdef · b1afdef
1 parent 01219ae
commit b1afdef
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/main/java/org/opensearch/knn/quantization/quantizer/QuantizerHelper.java b/src/main/java/org/opensearch/knn/quantization/quantizer/QuantizerHelper.java
@@ -57,6 +57,7 @@ static OneBitScalarQuantizationState calculateQuantizationState(
         }
 
         // Calculate below and above threshold means
+        trainingRequest.resetVectorValues();
         Pair<float[], float[]> belowAboveMeans = calculateBelowAboveThresholdMeans(trainingRequest, meanThresholds, sampledIndices);
         float[] belowThresholdMeans = belowAboveMeans.getA();
         float[] aboveThresholdMeans = belowAboveMeans.getB();

diff --git a/...rg/opensearch/knn/index/codec/KNN990Codec/NativeEngines990KnnVectorsWriterFlushTests.java b/...rg/opensearch/knn/index/codec/KNN990Codec/NativeEngines990KnnVectorsWriterFlushTests.java
@@ -730,7 +730,7 @@ public void testFlush_whenQuantizationIsProvided_whenBuildGraphDatStructureThres
             final Long expectedTimesGetVectorValuesIsCalled = vectorsPerField.stream().filter(Predicate.not(Map::isEmpty)).count();
             knnVectorValuesFactoryMockedStatic.verify(
                 () -> KNNVectorValuesFactory.getVectorValues(any(VectorDataType.class), any(DocsWithFieldSet.class), any()),
-                times(Math.toIntExact(expectedTimesGetVectorValuesIsCalled) * 2)
+                times(Math.toIntExact(expectedTimesGetVectorValuesIsCalled))
             );
         }
     }
@@ -832,7 +832,7 @@ public void testFlush_whenQuantizationIsProvided_whenBuildGraphDatStructureThres
             final Long expectedTimesGetVectorValuesIsCalled = vectorsPerField.stream().filter(Predicate.not(Map::isEmpty)).count();
             knnVectorValuesFactoryMockedStatic.verify(
                 () -> KNNVectorValuesFactory.getVectorValues(any(VectorDataType.class), any(DocsWithFieldSet.class), any()),
-                times(Math.toIntExact(expectedTimesGetVectorValuesIsCalled) * 2)
+                times(Math.toIntExact(expectedTimesGetVectorValuesIsCalled))
             );
         }
     }