In [plot_attr.py](https://github.com/SquareResearchCenter-AI/BEExAI/blob/main/src/beexai/explanation/plot_attr.py) - for **bar_plt (25-26)** and **plot_waterfall (55-56)** functions: `attributions.ndim > 1` condition is sufficient to apply mean operation. - Adjust the figure legend depending on whether we are displaying individual scores or the mean scores across the entire dataset.
In plot_attr.py
attributions.ndim > 1condition is sufficient to apply mean operation.