fix: fix sklearn base import for scikit learn v1.2

numaproj · Dec 21, 2022 · e0e2d77 · e0e2d77
1 parent 64ad0ed
commit e0e2d77
Show file tree

Hide file tree

Showing 3 changed files with 852 additions and 890 deletions.
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