From e79c3fa1e047d015fee37ce5432396af800e1e06 Mon Sep 17 00:00:00 2001 From: "Corey J. Nolet" Date: Tue, 5 Dec 2023 14:48:16 -0500 Subject: [PATCH 1/2] Fixing small bug --- python/raft-ann-bench/src/raft-ann-bench/run/__main__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 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 4611f39264..78adcb39aa 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 @@ -154,6 +154,7 @@ def run_build_and_search( ): for executable, ann_executable_path, algo in executables_to_run.keys(): # Need to write temporary configuration + print("ALGO: %s" % algo) temp_conf_filename = f"{conf_filename}_{algo}_{uuid.uuid1()}.json" with open(temp_conf_filename, "w") as f: temp_conf = dict() @@ -574,8 +575,8 @@ def add_algo_group(group_list): index["search_params"].append(search_dict) executables_to_run[executable]["index"].append(index) - if len(index["search_params"]) == 0: - print("No search parameters were added to configuration") + if len(index["search_params"]) == 0: + print("No search parameters were added to configuration") run_build_and_search( conf_file, From 521aced0bd7cd7c22f120703191fc0806939d08d Mon Sep 17 00:00:00 2001 From: "Corey J. Nolet" Date: Tue, 5 Dec 2023 16:34:51 -0500 Subject: [PATCH 2/2] Removing unecessary print --- python/raft-ann-bench/src/raft-ann-bench/run/__main__.py | 1 - 1 file changed, 1 deletion(-) 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 78adcb39aa..9841b47b98 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 @@ -154,7 +154,6 @@ def run_build_and_search( ): for executable, ann_executable_path, algo in executables_to_run.keys(): # Need to write temporary configuration - print("ALGO: %s" % algo) temp_conf_filename = f"{conf_filename}_{algo}_{uuid.uuid1()}.json" with open(temp_conf_filename, "w") as f: temp_conf = dict()