From 85d9d8430fd1abd3aa50390d14014f5e89a92ad3 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Wed, 15 May 2024 11:04:14 -0700 Subject: [PATCH] aider: Color the "gpt-3.5" models green in the plotting code. # Aider chat conversation: USER: color the gpt-3.5 models a new color ASSISTANT: To color the "gpt-3.5" models a new color, we need to modify the plotting code in `benchmark/over_time.py` to check for "gpt-3.5" in the model name and assign a different color. Here are the changes: benchmark/over_time.py ```python <<<<<<< SEARCH colors = ['red' if 'gpt-4' in model else 'blue' for model in models] ======= colors = ['red' if 'gpt-4' in model else 'green' if 'gpt-3.5' in model else 'blue' for model in models] >>>>>>> REPLACE ``` --- benchmark/over_time.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmark/over_time.py b/benchmark/over_time.py index 44dcb0924..7083208ce 100644 --- a/benchmark/over_time.py +++ b/benchmark/over_time.py @@ -17,7 +17,7 @@ def plot_over_time(yaml_file): models.append(entry['model']) plt.figure(figsize=(10, 6)) - colors = ['red' if 'gpt-4' in model else 'blue' for model in models] + colors = ['red' if 'gpt-4' in model else 'green' if 'gpt-3.5' in model else 'blue' for model in models] plt.scatter(dates, pass_rates, c=colors, alpha=0.5) for i, model in enumerate(models):