feat: add new color for all "-4o-" models except "gpt-4o-mini"

This commit is contained in:
Paul Gauthier (aider) 2024-08-14 06:22:48 -07:00
parent 1f59687e9d
commit c4f70d81b7

View file

@ -50,7 +50,10 @@ def plot_over_time(yaml_file):
spine.set_linewidth(0.5)
colors = [
"red" if "gpt-4" in model else "green" if "gpt-3.5" in model else "blue" for model in models
"orange" if "-4o-" in model and "gpt-4o-mini" not in model
else "red" if "gpt-4" in model
else "green" if "gpt-3.5" in model
else "blue" for model in models
]
ax.scatter(dates, pass_rates, c=colors, alpha=0.5, s=120)