Added a note at the bottom of the graph in swe_bench_lite.py.

This commit is contained in:
Paul Gauthier 2024-05-31 07:44:27 -07:00
parent ffec28cc15
commit 8bcebfa475

View file

@ -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.