From ddd65dc197529b4c6fb296f8fcebde44b2ddfa5d Mon Sep 17 00:00:00 2001 From: Sarah Alnegheimish Date: Mon, 18 Mar 2024 17:12:37 -0400 Subject: [PATCH] remove sklearn base --- cardea/primitives/processing/categorizer.py | 4 ++-- setup.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/cardea/primitives/processing/categorizer.py b/cardea/primitives/processing/categorizer.py index 3cb5a30b..0504aa0c 100644 --- 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 index 5abf1065..fffa6e64 100644 --- 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',