From 8bcebfa4754f8966a9844ac29c5b068c4058248e Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Fri, 31 May 2024 07:44:27 -0700 Subject: [PATCH] Added a note at the bottom of the graph in swe_bench_lite.py. --- benchmark/swe_bench_lite.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/benchmark/swe_bench_lite.py b/benchmark/swe_bench_lite.py index a84c238b0..e0546a531 100644 --- a/benchmark/swe_bench_lite.py +++ b/benchmark/swe_bench_lite.py @@ -77,6 +77,19 @@ def plot_swe_bench_lite(data_file): fontsize=16, color=font_color, ) + + + # Add note at the bottom of the graph + note = "Note: (570) and (2294) refer to the number of instances that were processed by the agent." + plt.figtext( + 0.5, 0.025, + note, + wrap=True, + horizontalalignment='center', + fontsize=12, + color=font_color, + ) + plt.tight_layout(pad=3.0) out_fname = Path(data_file) @@ -85,10 +98,6 @@ def plot_swe_bench_lite(data_file): imgcat(fig) ax.xaxis.label.set_color(font_color) - # Add note at the bottom of the graph - note = "(570) and (2294) refer to the number of instances that were processed by the agent." - plt.figtext(0.5, 0.01, note, wrap=True, horizontalalignment='center', fontsize=10, color=font_color) - # Note: (570) and (2294) refer to the number of instances that were processed by the agent.