dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-29 16:54:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

aider: Set bar color/fill style for each edit-format in show_stats function using a color map.

Browse source 
# 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:
Paul Gauthier 2023-06-29 20:56:58 -07:00
parent a02cadbc36
commit 73f99996df
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
benchmark/benchmark.py
View file

@ -69,7 +69,8 @@ def show_stats(dirnames):
df_grouped2 = df.groupby(["model", "edit_format"])["pass_rate_2"].mean() df_grouped2 = df.groupby(["model", "edit_format"])["pass_rate_2"].mean()
fig, ax = plt.subplots(figsize=(10, 6)) 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_xlabel("Percent of passed unittests")
ax.set_ylabel("Model") ax.set_ylabel("Model")