[BUG] fix #151 (#153)

* BUG FIX: #151 * MODIFY latest * Codespell
juaml · Jun 16, 2022 · aab4260 · aab4260
1 parent 0f45481
commit aab4260
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/docs/changes/latest.inc b/docs/changes/latest.inc
@@ -58,6 +58,8 @@ Bugs
 
 - Fix Bug in the transformer wrapper implementation (:gh:`122` by `Sami Hamdan`_).
 
+- Fix Bug Target Transformer missing BaseEstimator (:gh:`151` by `Sami Hamdan`_).
+
 API changes
 ~~~~~~~~~~~
 

diff --git a/julearn/transformers/target.py b/julearn/transformers/target.py
@@ -3,12 +3,12 @@
 # License: AGPL
 from inspect import signature
 import pandas as pd
-from sklearn.base import TransformerMixin
+from sklearn.base import TransformerMixin, BaseEstimator
 
 from .. utils import raise_error
 
 
-class TargetTransfromerWrapper(TransformerMixin):
+class TargetTransfromerWrapper(TransformerMixin, BaseEstimator):
 
     def __init__(self, transformer, **params):
         """Using a sklearn transformer and applying them to the target/y.