fix: the HNSW index doesn't respect to refine factor (#2122)

Signed-off-by: BubbleCal <[email protected]>
lancedb · Mar 31, 2024 · 2f4ed88 · 2f4ed88
1 parent 874bf9c
commit 2f4ed88
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/rust/lance/src/index/vector/hnsw.rs b/rust/lance/src/index/vector/hnsw.rs
@@ -163,10 +163,11 @@ impl VectorIndex for HNSWIndex {
             )
         };
 
-        let ef = query.k + query.k / 2;
+        let k = query.k * query.refine_factor.unwrap_or(1) as usize;
+        let ef = k + k / 2;
         let results = self.hnsw.search(
             query.key.as_primitive::<Float32Type>().as_slice(),
-            query.k,
+            k,
             ef,
             bitmap,
         )?;