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
…828)


Signed-off-by: Jinfeng <[email protected]>
  • Loading branch information
lijinf2 authored Jan 24, 2025
1 parent 020975c commit d770bd1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/tests/test_approximate_nearest_neighbors.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright (c) 2024, NVIDIA CORPORATION.
# Copyright (c) 2024-2025, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand Down 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 d770bd1

Please sign in to comment.