From 7ee3c27a18d54d53b61e70b766f369ddfe8d6d48 Mon Sep 17 00:00:00 2001 From: Jinfeng Date: Fri, 24 Jan 2025 19:16:30 +0000 Subject: [PATCH] relax tolerance per ivf_flat is unstable with default None algoParam Signed-off-by: Jinfeng --- python/tests/test_approximate_nearest_neighbors.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tests/test_approximate_nearest_neighbors.py b/python/tests/test_approximate_nearest_neighbors.py index f3bb8fd8..f86ebaa5 100644 --- a/python/tests/test_approximate_nearest_neighbors.py +++ b/python/tests/test_approximate_nearest_neighbors.py @@ -627,7 +627,7 @@ def test_ivfflat( # cuvs ivf_flat None sets nlist to 1000 and nprobe to 20, leading to unstable results when run multiple times expected_avg_recall: float = 0.95 if algoParams != None else 0.5 expected_avg_dist_gap: float = 1e-4 if algoParams != None else 1e-2 - tolerance: float = 1e-4 if algoParams != None else 1e-2 + tolerance: float = 1e-4 if algoParams != None else 0.05 data_shape: Tuple[int, int] = (10000, 50) ann_algorithm_test_func( combo=combo,