style fixes

rapidsai · Nov 7, 2023 · a195530 · a195530
1 parent 16d619e
commit a195530
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/raft-ann-bench/src/raft-ann-bench/data_export/__main__.py b/python/raft-ann-bench/src/raft-ann-bench/data_export/__main__.py
@@ -114,14 +114,15 @@ def convert_json_to_csv_search(dataset, dataset_path):
                         write.iloc[s_index, write_ncols] = build_df.iloc[
                             b_index, 2
                         ]
-                        write.iloc[s_index, write_ncols + 1:] = build_df.iloc[
+                        write.iloc[s_index, write_ncols + 1 :] = build_df.iloc[
                             b_index, 3:
                         ]
                         break
         else:
             warnings.warn(
                 f"Build CSV not found for {algo_name}, build params won't be "
-                "appended in the Search CSV")
+                "appended in the Search CSV"
+            )
 
         write.to_csv(file.replace(".json", ".csv"), index=False)