[BUG] cuvs-bench data_export fixes

python/cuvs_bench/cuvs_bench/run/data_export.py

@@ -217,12 +217,16 @@ def convert_json_to_csv_search(dataset, dataset_path):
                 write["build time"] = None
                 write["build threads"] = None
                 write["build cpu_time"] = None
-                write["build GPU"] = None
 
-                for col_idx in range(6, len(build_df.columns)):
+                start_idx = 5
+                if "GPU" in build_df.columns:
+                    start_idx = 6
+                    write["build GPU"] = None
+                for col_idx in range(start_idx, len(build_df.columns)):
                     col_name = build_df.columns[col_idx]
                     write[col_name] = None
-
+                    if col_name == "num_threads":
+                        write["build_num_threads"] = None
                 for s_index, search_row in write.iterrows():
                     for b_index, build_row in build_df.iterrows():
                         if search_row["index_name"] == build_row["index_name"]:

python/cuvs_bench/cuvs_bench/tests/test_cli.py

@@ -170,7 +170,7 @@ def test_run_command_creates_results(temp_datasets_dir: Path):
                 "build time",
                 "build threads",
                 "build cpu_time",
-                "build GPU",
+                "ef_construction",
                 "graph_degree",
                 "intermediate_graph_degree",
                 "label",