fix style

rapidsai · Dec 15, 2023 · b6534c3 · b6534c3
1 parent fe14bbc
commit b6534c3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/python/raft-ann-bench/src/raft-ann-bench/plot/__main__.py b/python/raft-ann-bench/src/raft-ann-bench/plot/__main__.py
@@ -296,8 +296,10 @@ def mean_y(algo):
     ax = df.plot.bar(rot=0, color=colors)
     fig = ax.get_figure()
     print(f"writing build output to {fn_out}")
-    plt.title("Average Build Time within Recall Range "
-              f"for k={k} batch_size={batch_size}")
+    plt.title(
+        "Average Build Time within Recall Range "
+        f"for k={k} batch_size={batch_size}"
+    )
     plt.suptitle(f"{dataset}")
     plt.ylabel("Build Time (s)")
     fig.savefig(fn_out)
@@ -613,7 +615,7 @@ def main():
             build_output_filepath,
             args.dataset,
             k,
-            batch_size
+            batch_size,
         )