Don't remove build json when it fails if we have to search next

rapidsai · Nov 8, 2023 · a72d9ca · a72d9ca
1 parent 0d28f73
commit a72d9ca
Showing 1 changed file with 2 additions and 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
@@ -130,7 +130,8 @@ def run_build_and_search(
                 except Exception as e:
                     print("Error occurred running benchmark: %s" % e)
                 finally:
-                    os.remove(temp_conf_filename)
+                    if not search:
+                        os.remove(temp_conf_filename)
 
         if search:
             search_folder = os.path.join(legacy_result_folder, "search")