From 4cda43f4b258778eb899d7a2ba38b1e260e00db1 Mon Sep 17 00:00:00 2001 From: yaommen Date: Wed, 25 Mar 2026 01:15:08 +0800 Subject: [PATCH] test: stabilize distributed IVF query results --- rust/lance/src/index/vector/ivf/v2.rs | 2 ++ 1 file changed, 2 insertions(+) diff --git a/rust/lance/src/index/vector/ivf/v2.rs b/rust/lance/src/index/vector/ivf/v2.rs index 92b0549f43b..92ddea55cbc 100644 --- a/rust/lance/src/index/vector/ivf/v2.rs +++ b/rust/lance/src/index/vector/ivf/v2.rs @@ -1848,6 +1848,7 @@ mod tests { .unwrap() .nearest("vector", q.as_ref(), K) .unwrap() + .minimum_nprobes(TWO_FRAG_NUM_PARTITIONS) .try_into_batch() .await .unwrap(); @@ -1984,6 +1985,7 @@ mod tests { .unwrap() .nearest("vector", q.as_ref(), K) .unwrap() + .minimum_nprobes(TWO_FRAG_NUM_PARTITIONS) .try_into_batch() .await .unwrap();