feat: Replace orange color with purple for "-4o" models

This commit is contained in:
Paul Gauthier (aider) 2024-08-14 06:29:13 -07:00
parent fb0b348bec
commit d2b4846b95

View file

@ -51,7 +51,7 @@ def plot_over_time(yaml_file):
colors = [ colors = [
( (
"orange" "purple"
if "-4o" in model and "gpt-4o-mini" not in model 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" else "red" if "gpt-4" in model else "green" if "gpt-3.5" in model else "blue"
) )
@ -59,14 +59,14 @@ def plot_over_time(yaml_file):
] ]
# Separate data points by color # Separate data points by color
orange_points = [(d, r) for d, r, c in zip(dates, pass_rates, colors) if c == "orange"] purple_points = [(d, r) for d, r, c in zip(dates, pass_rates, colors) if c == "purple"]
red_points = [(d, r) for d, r, c in zip(dates, pass_rates, colors) if c == "red"] red_points = [(d, r) for d, r, c in zip(dates, pass_rates, colors) if c == "red"]
green_points = [(d, r) for d, r, c in zip(dates, pass_rates, colors) if c == "green"] green_points = [(d, r) for d, r, c in zip(dates, pass_rates, colors) if c == "green"]
# Plot lines for orange, red, and green points # Plot lines for purple, red, and green points
if orange_points: if purple_points:
orange_dates, orange_rates = zip(*sorted(orange_points)) purple_dates, purple_rates = zip(*sorted(purple_points))
ax.plot(orange_dates, orange_rates, c="orange", alpha=0.5, linewidth=1) ax.plot(purple_dates, purple_rates, c="purple", alpha=0.5, linewidth=1)
if red_points: if red_points:
red_dates, red_rates = zip(*sorted(red_points)) red_dates, red_rates = zip(*sorted(red_points))
ax.plot(red_dates, red_rates, c="red", alpha=0.5, linewidth=1) ax.plot(red_dates, red_rates, c="red", alpha=0.5, linewidth=1)