[plot] Optional plot titles

benchmark.py

@@ -95,6 +95,12 @@ def parse_command_line_arguments():
         "settings",
         help='settings to compare solvers on (e.g. "high_accuracy")',
     )
+    parser_plot.add_argument(
+        "--linewidth",
+        help="width of plotted lines in px",
+        type=int,
+        default=3,
+    )
     parser_plot.add_argument(
         "--savefig",
         help="path to a file to save the plot to (rather than displaying it)",
@@ -105,10 +111,8 @@ def parse_command_line_arguments():
         nargs="+",
     )
     parser_plot.add_argument(
-        "--linewidth",
-        help="width of plotted lines in px",
-        type=int,
-        default=3,
+        "--title",
+        help='plot title (set to "" to disable)',
     )
 
     # report
@@ -250,6 +254,7 @@ def load_test_set(name: str) -> TestSet:
             solvers=args.solvers,
             linewidth=args.linewidth,
             savefig=args.savefig,
+            title=args.title,
         )
 
     if args.command in ["report", "run"]:

qpsolvers_benchmark/plot_metric.py

@@ -34,6 +34,7 @@ def plot_metric(
     solvers: Optional[List[str]] = None,
     linewidth: float = 3.0,
     savefig: Optional[str] = None,
+    title: Optional[str] = None,
 ) -> None:
     """Plot comparing solvers on a given metric.
 
@@ -45,6 +46,7 @@ def plot_metric(
         solvers: Names of solvers to compare (default: all).
         linewidth: Width of output lines, in px.
         savefig: If set, save plot to this path rather than displaying it.
+        title: Plot title, set to "" to disable.
     """
     assert issubclass(df[metric].dtype.type, np.floating)
     nb_problems = test_set.count_problems()
@@ -68,9 +70,12 @@ def plot_metric(
         padded_y = np.hstack([y, [nb_solved]])
         plt.step(padded_values, padded_y, linewidth=linewidth)
     plt.legend(plot_solvers)
-    plt.title(
-        f"Comparing {metric} on {test_set.title} with {settings} settings"
-    )
+    if title is None:
+        plt.title(
+            f"Comparing {metric} on {test_set.title} with {settings} settings"
+        )
+    elif title != "":
+        plt.title(title)
     plt.xlabel(metric)
     plt.xscale("log")
     plt.axhline(y=nb_problems, color="gray", linestyle=":")