ADD BaseEstimator to TargetConfoundRemover #151

juaml · Aug 18, 2022 · 728cfaf · 728cfaf
1 parent d9ae892
commit 728cfaf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/docs/changes/latest.inc b/docs/changes/latest.inc
@@ -19,6 +19,9 @@ Enhancements
 Bugs
 ~~~~
 
+
+- ADD BaseEstimator to ConfoundRemoval for Target  (:gh:`151` by `Sami Hamdan`_).
+
 API changes
 ~~~~~~~~~~~
 

diff --git a/julearn/transformers/confounds.py b/julearn/transformers/confounds.py
@@ -183,7 +183,7 @@ def _apply_threshold(self, residuals):
         return residuals
 
 
-class TargetConfoundRemover(TransformerMixin):
+class TargetConfoundRemover(BaseEstimator, TransformerMixin):
 
     def __init__(self,
                  model_confound=None,