From 84322b452a6d88277c03a4bf084a3733a1e789d1 Mon Sep 17 00:00:00 2001 From: monte-flora Date: Thu, 2 Apr 2026 12:14:31 +0000 Subject: [PATCH] Fix plot_correlated_features for interaction methods (#6) Guard _add_correlated_brackets to only run for single-variable methods (SINGLE_VAR_METHODS). Interaction methods (hstat, perm_based, ale_variance_interactions) have pair names like "CAPE__Shear" that can't be looked up in the correlation matrix. Closes #6 Co-Authored-By: Claude Opus 4.6 (1M context) --- skexplain/plot/plot_permutation_importance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)),