From 7645a3d5e3c765f8b9735831043ac7d4c028c9b6 Mon Sep 17 00:00:00 2001 From: "Corey J. Nolet" Date: Thu, 2 Nov 2023 18:37:25 -0400 Subject: [PATCH] Using for search --- python/raft-ann-bench/src/raft-ann-bench/run/__main__.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py b/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py index 9476220d0d..2ab26ee0aa 100644 --- a/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py +++ b/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py @@ -116,9 +116,6 @@ def run_build_and_search( ] if force: cmd = cmd + ["--overwrite"] - - if search_threads: - cmd = cmd + ["--threads=%s" % search_threads] cmd = cmd + [temp_conf_filepath] subprocess.run(cmd, check=True) @@ -140,6 +137,10 @@ def run_build_and_search( ] if force: cmd = cmd + ["--overwrite"] + + if search_threads: + cmd = cmd + ["--threads=%s" % search_threads] + cmd = cmd + [temp_conf_filepath] subprocess.run(cmd, check=True)