This commit is contained in:
Paul Gauthier 2024-05-23 10:03:43 -07:00
parent 5ff4deb12e
commit c0f5cbb9b5
3 changed files with 31 additions and 37 deletions

View file

@ -42,7 +42,6 @@ def plot_swe_bench_lite(data_file):
yval = bar.get_height()
y = yval + 0.75 if "Aider" in model else yval - 1.25
va = "bottom" if "Aider" in model else "top"
fontweight = 800 if "Aider" in model else "normal"
ax.text(
bar.get_x() + bar.get_width() / 2,
@ -51,21 +50,16 @@ def plot_swe_bench_lite(data_file):
ha="center",
va=va,
fontsize=14,
fontweight=fontweight,
)
# ax.set_xlabel("Models", fontsize=18)
ax.set_ylabel("Instances resolved (%)", fontsize=18, color=font_color)
ax.set_title("SWE Bench Lite", fontsize=20)
ax.set_ylim(0, 29.9)
xticks = plt.xticks(
plt.xticks(
fontsize=16,
color=font_color,
)
for label in xticks[1]:
if "Aider" in label.get_text():
label.set_fontweight(800)
label.set_fontsize(16)
plt.tight_layout(pad=3.0)
plt.savefig("swe_bench_lite.jpg")
plt.savefig("swe_bench_lite.svg")