remove sklearn base

MLBazaar · Mar 18, 2024 · ddd65dc · ddd65dc
1 parent 9046d1b
commit ddd65dc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cardea/primitives/processing/categorizer.py b/cardea/primitives/processing/categorizer.py
@@ -1,9 +1,8 @@
 import numpy as np
 import pandas as pd
-from sklearn.base import TransformerMixin
 
 
-class Categorizer(TransformerMixin):
+class Categorizer():
     """Transform categorical values into numerical codes.
     """
 
@@ -19,6 +18,7 @@ def fit(self, y):
     def transform(self, y):
         if y is None:
             return y
+
         return pd.Categorical(y, categories=self.categories).codes
 
     def inverse_transform(self, y_codes):

diff --git a/setup.py b/setup.py
@@ -33,6 +33,7 @@
     'bumpversion>=0.5.3',
     'pip>=9.0.1',
     'watchdog>=0.8.3',
+    'jedi<0.18',
 
     # docs
     'm2r2>=0.2.5,<0.3',