fix: sklearn base import for scikit learn v1.2 (#112)

Signed-off-by: Avik Basu <avikbasu93@gmail.com>
numaproj · Dec 21, 2022 · df20591 · df20591
1 parent 64ad0ed
commit df20591
Show file tree

Hide file tree

Showing 4 changed files with 853 additions and 891 deletions.
diff --git a/Makefile b/Makefile
@@ -24,7 +24,7 @@ lint: format
 
 # install all dependencies
 setup:
-	poetry install --all-extras
+	poetry install --with dev --all-extras
 
 # test your application (tests in the tests/ directory)
 test:

diff --git a/numalogic/preprocess/transformer.py b/numalogic/preprocess/transformer.py
@@ -2,12 +2,12 @@
 
 import numpy as np
 from numpy.typing import ArrayLike
-from sklearn.base import BaseEstimator, TransformerMixin, _OneToOneFeatureMixin
+from sklearn.base import BaseEstimator, TransformerMixin
 
 LOGGER = logging.getLogger(__name__)
 
 
-class _DataIndependentTransformers(_OneToOneFeatureMixin, TransformerMixin, BaseEstimator):
+class _DataIndependentTransformers(TransformerMixin, BaseEstimator):
     def fit(self, _: ArrayLike):
         return self