Skip to content

Commit

Permalink
relax tolerance per ivf_flat is unstable with default None algoParam
Browse files Browse the repository at this point in the history
Signed-off-by: Jinfeng <[email protected]>
  • Loading branch information
lijinf2 committed Jan 24, 2025
1 parent 020975c commit 7ee3c27
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/tests/test_approximate_nearest_neighbors.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 7ee3c27

Please sign in to comment.