Use Math.ceil instead of rounding

dungba88 · Nov 18, 2024 · f9da336 · f9da336
1 parent ccd3e25
commit f9da336
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lucene/core/src/java/org/apache/lucene/search/TwoPhaseKnnVectorQuery.java b/lucene/core/src/java/org/apache/lucene/search/TwoPhaseKnnVectorQuery.java
@@ -51,7 +51,7 @@ public class TwoPhaseKnnVectorQuery extends KnnFloatVectorQuery {
    */
   public TwoPhaseKnnVectorQuery(
       String field, float[] target, int k, double oversample, Query filter) {
-    super(field, target, k + (int) Math.round(k * oversample), filter);
+    super(field, target, k + (int) Math.ceil(k * oversample), filter);
     if (oversample < 0) {
       throw new IllegalArgumentException("oversample must be non-negative, got " + oversample);
     }