From 73f99996dffdb97f102c9210522202cb2bb3554f Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Thu, 29 Jun 2023 20:56:58 -0700 Subject: [PATCH] aider: Set bar color/fill style for each edit-format in `show_stats` function using a color map. # Context: USER: how can i set the bar color/fill style for each edit-format ASSISTANT: To set the bar color/fill style for each edit-format, you need to modify the `show_stats` function in the `benchmark/benchmark.py` file. Here are the steps: 1. Define a dictionary mapping each edit-format to a specific color or fill style. 2. Use the `color` or `fill` parameter in the `plot` function to set the color or fill style for each bar. Here is the code change: ```python benchmark/benchmark.py <<<<<<< ORIGINAL fig, ax = plt.subplots(figsize=(10, 6)) df_grouped2.unstack().plot(kind="barh", ax=ax) ======= fig, ax = plt.subplots(figsize=(10, 6)) color_map = {"edit_format1": "blue", "edit_format2": "green", "edit_format3": "red"} # Define your color map here df_grouped2.unstack().plot(kind="barh", ax=ax, color=[color_map[i] for i in df_grouped2.index]) >>>>>>> UPDATED ``` Please replace `"edit_format1"`, `"edit_format2"`, and `"edit_format3"` with your actual edit formats, and `"blue"`, `"green"`, and `"red"` with your desired colors. --- benchmark/benchmark.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/benchmark/benchmark.py b/benchmark/benchmark.py index cb264fe68..599502372 100755 --- a/benchmark/benchmark.py +++ b/benchmark/benchmark.py @@ -69,7 +69,8 @@ def show_stats(dirnames): df_grouped2 = df.groupby(["model", "edit_format"])["pass_rate_2"].mean() fig, ax = plt.subplots(figsize=(10, 6)) - df_grouped2.unstack().plot(kind="barh", ax=ax) + color_map = {"edit_format1": "blue", "edit_format2": "green", "edit_format3": "red"} # Define your color map here + df_grouped2.unstack().plot(kind="barh", ax=ax, color=[color_map[i] for i in df_grouped2.index]) ax.set_xlabel("Percent of passed unittests") ax.set_ylabel("Model")