mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-28 08:14:59 +00:00
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.
This commit is contained in:
parent
a02cadbc36
commit
73f99996df
1 changed files with 2 additions and 1 deletions
|
@ -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")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue