remove TransformerMixin inheritance

mind-inria · Oct 16, 2024 · f946696 · f946696
1 parent bdb0e5f
commit f946696
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/hidimstat/cpi.py b/hidimstat/cpi.py
@@ -1,10 +1,10 @@
 import numpy as np
 from joblib import Parallel, delayed
-from sklearn.base import BaseEstimator, TransformerMixin, check_is_fitted, clone
+from sklearn.base import BaseEstimator, check_is_fitted, clone
 from sklearn.metrics import root_mean_squared_error
 
 
-class CPI(BaseEstimator, TransformerMixin):
+class CPI(BaseEstimator):
     """
     Conditional Permutation Importance (CPI) algorithm.
     :footcite:t:`Chamma_NeurIPS2023` and for group-level see

diff --git a/hidimstat/loco.py b/hidimstat/loco.py
@@ -1,10 +1,10 @@
 import numpy as np
 from joblib import Parallel, delayed
-from sklearn.base import BaseEstimator, TransformerMixin, check_is_fitted, clone
+from sklearn.base import BaseEstimator, check_is_fitted, clone
 from sklearn.metrics import root_mean_squared_error
 
 
-class LOCO(BaseEstimator, TransformerMixin):
+class LOCO(BaseEstimator):
     """
     Leave-One-Covariate-Out (LOCO) algorithm as described in
     :footcite:t:`Chamma_NeurIPS2023`.

diff --git a/hidimstat/permutation_importance.py b/hidimstat/permutation_importance.py
@@ -1,10 +1,10 @@
 import numpy as np
 from joblib import Parallel, delayed
-from sklearn.base import BaseEstimator, TransformerMixin, check_is_fitted
+from sklearn.base import BaseEstimator, check_is_fitted
 from sklearn.metrics import root_mean_squared_error
 
 
-class PermutationImportance(BaseEstimator, TransformerMixin):
+class PermutationImportance(BaseEstimator):
     """