Fixing small bug

rapidsai · Dec 5, 2023 · e79c3fa · e79c3fa
1 parent 8e1c62c
commit e79c3fa
Showing 1 changed file with 3 additions and 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
@@ -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,