mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 00:35:00 +00:00
d37bace5d7
# Aider chat conversation:
USER: make the percentages on top of aiders bars bold. make the aider x labels bold too
ASSISTANT: To make the percentages on top of Aider's bars bold and the Aider x labels bold, we need to modify the text properties in the plotting section.
Here are the *SEARCH/REPLACE* blocks:
benchmark/swe_bench_lite.py
```python
<<<<<<< SEARCH
if "Aider" in model:
y = yval + 0.75
va = "bottom"
else:
y = yval - 1.25
va = "top"
ax.text(
bar.get_x() + bar.get_width() / 2,
y,
f"{yval}%",
ha="center",
va=va,
fontsize=14,
# alpha=0.75,
)
=======
y = yval + 0.75 if "Aider" in model else yval - 1.25
va = "bottom" if "Aider" in model else "top"
fontweight = "bold" if "Aider" in model else "normal"
ax.text(
bar.get_x() + bar.get_width() / 2,
y,
f"{yval}%",
ha="center",
va=va,
fontsize=14,
fontweight=fontweight,
)
>>>>>>> REPLACE
```
benchmark/swe_bench_lite.py
```python
<<<<<<< SEARCH
plt.xticks(
# rotation=45,
# ha="center",
fontsize=16,
color=font_color,
)
=======
xticks = plt.xticks(
fontsize=16,
color=font_color,
)
for label in xticks[1]:
if "Aider" in label.get_text():
label.set_fontweight("bold")
>>>>>>> REPLACE
```
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
import matplotlib.pyplot as plt
|
|
from imgcat import imgcat
|
|
from matplotlib import rc
|
|
|
|
|
|
def plot_swe_bench_lite(data_file):
|
|
with open(data_file, "r") as file:
|
|
lines = file.readlines()
|
|
|
|
models = []
|
|
pass_rates = []
|
|
|
|
for line in lines:
|
|
if line.strip():
|
|
pass_rate, model = line.split("%")
|
|
model = model.strip()
|
|
model = model.replace("|", "\n")
|
|
models.append(model.strip())
|
|
pass_rates.append(float(pass_rate.strip()))
|
|
|
|
plt.rcParams["hatch.linewidth"] = 0.5
|
|
plt.rcParams["hatch.color"] = "#444444"
|
|
|
|
font_color = "#555"
|
|
rc("font", **{"family": "sans-serif", "sans-serif": ["Helvetica"], "size": 10})
|
|
plt.rcParams["text.color"] = font_color
|
|
|
|
fig, ax = plt.subplots(figsize=(10, 5))
|
|
ax.grid(axis="y", zorder=0, lw=0.2)
|
|
for spine in ax.spines.values():
|
|
spine.set_edgecolor("#DDDDDD")
|
|
spine.set_linewidth(0.5)
|
|
|
|
colors = ["#17965A" if "Aider" in model else "#b3d1e6" for model in models]
|
|
bars = []
|
|
for model, pass_rate, color in zip(models, pass_rates, colors):
|
|
alpha = 0.6 if "Aider" in model else 0.3
|
|
bar = ax.bar(model, pass_rate, color=color, alpha=alpha, zorder=3)
|
|
bars.append(bar[0])
|
|
|
|
for model, bar in zip(models, bars):
|
|
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 = "bold" if "Aider" in model else "normal"
|
|
|
|
ax.text(
|
|
bar.get_x() + bar.get_width() / 2,
|
|
y,
|
|
f"{yval}%",
|
|
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(
|
|
fontsize=16,
|
|
color=font_color,
|
|
)
|
|
for label in xticks[1]:
|
|
if "Aider" in label.get_text():
|
|
label.set_fontweight("bold")
|
|
plt.tight_layout(pad=3.0)
|
|
plt.savefig("swe_bench_lite.jpg")
|
|
plt.savefig("swe_bench_lite.svg")
|
|
imgcat(fig)
|
|
ax.xaxis.label.set_color(font_color)
|
|
|
|
|
|
# Example usage
|
|
plot_swe_bench_lite("benchmark/tmp.txt")
|