Merge pull request #779 from Marchlak/master

fix (#742)
mljar · Sep 25, 2024 · e5fd66e · e5fd66e
2 parents ecd68f6 + 0312ae3
commit e5fd66e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/supervised/preprocessing/preprocessing_categorical.py b/supervised/preprocessing/preprocessing_categorical.py
@@ -1,4 +1,5 @@
 import numpy as np
+import pandas as pd
 
 from supervised.preprocessing.label_binarizer import LabelBinarizer
 from supervised.preprocessing.label_encoder import LabelEncoder
@@ -78,6 +79,8 @@ def transform(self, X):
                     if transformed_values.dtype != X.loc[:, column].dtype and \
                         (X.loc[:, column].dtype == bool or X.loc[:, column].dtype == int):
                         X = X.astype({column: transformed_values.dtype})
+                    if isinstance(X[column].dtype, pd.CategoricalDtype):
+                        X[column] = X[column].astype('object')
                     X.loc[:, column] = transformed_values
 
             return X