diff --git a/skexplain/plot/plot_permutation_importance.py b/skexplain/plot/plot_permutation_importance.py index 8592757..3a92193 100644 --- a/skexplain/plot/plot_permutation_importance.py +++ b/skexplain/plot/plot_permutation_importance.py @@ -297,7 +297,7 @@ def plot_variable_importance( zorder=2, ) - if plot_correlated_features: + if plot_correlated_features and method in self.SINGLE_VAR_METHODS: self._add_correlated_brackets( ax, np.arange(len(scores_to_plot)),