fix

Author: jona-sassenhagen

category_encoders/backward_difference.py

@@ -214,7 +214,7 @@ def transform(self, X, override_return_df=False):
         X = self.backward_difference_coding(X, mapping=self.mapping)
 
         if self.drop_invariant:
-            X = X.drop(columns=colself.drop_cols)
+            X = X.drop(columns=self.drop_cols)
 
         if self.return_df or override_return_df:
             return X

category_encoders/cat_boost.py

@@ -208,7 +208,7 @@ def transform(self, X, y=None, override_return_df=False):
         )
 
         if self.drop_invariant:
-            X = X.drop(columns=colself.drop_cols)
+            X = X.drop(columns=self.drop_cols)
 
         if self.return_df or override_return_df:
             return X

category_encoders/count.py

@@ -218,7 +218,7 @@ def transform(self, X, y=None, override_return_df=False):
         X, _ = self._transform_count_encode(X, y)
 
         if self.drop_invariant:
-            X = X.drop(columns=colself.drop_cols)
+            X = X.drop(columns=self.drop_cols)
 
         if self.return_df or override_return_df:
             return X

category_encoders/glmm.py

@@ -226,7 +226,7 @@ def transform(self, X, y=None, override_return_df=False):
         # Postprocessing
         # Note: We should not even convert these columns.
         if self.drop_invariant:
-            X = X.drop(columns=colself.drop_cols)
+            X = X.drop(columns=self.drop_cols)
 
         if self.return_df or override_return_df:
             return X

category_encoders/hashing.py

@@ -308,7 +308,7 @@ def _transform(self, X, override_return_df=False):
         X = self.hashing_trick(X, hashing_method=self.hash_method, N=self.n_components, cols=self.cols)
 
         if self.drop_invariant:
-            X = X.drop(columns=colself.drop_cols)
+            X = X.drop(columns=self.drop_cols)
 
         if self.return_df or override_return_df:
             return X

category_encoders/helmert.py

@@ -211,7 +211,7 @@ def transform(self, X, override_return_df=False):
         X = self.helmert_coding(X, mapping=self.mapping)
 
         if self.drop_invariant:
-            X = X.drop(columns=colself.drop_cols)
+            X = X.drop(columns=self.drop_cols)
 
         if self.return_df or override_return_df:
             return X

category_encoders/james_stein.py

@@ -289,7 +289,7 @@ def transform(self, X, y=None, override_return_df=False):
         # Postprocessing
         # Note: We should not even convert these columns.
         if self.drop_invariant:
-            X = X.drop(columns=colself.drop_cols)
+            X = X.drop(columns=self.drop_cols)
 
         if self.return_df or override_return_df:
             return X

category_encoders/leave_one_out.py

@@ -195,7 +195,7 @@ def transform(self, X, y=None, override_return_df=False):
         )
 
         if self.drop_invariant:
-            X = X.drop(columns=colself.drop_cols)
+            X = X.drop(columns=self.drop_cols)
 
         if self.return_df or override_return_df:
             return X

category_encoders/m_estimate.py

@@ -221,7 +221,7 @@ def transform(self, X, y=None, override_return_df=False):
         # Postprocessing
         # Note: We should not even convert these columns.
         if self.drop_invariant:
-            X = X.drop(columns=colself.drop_cols)
+            X = X.drop(columns=self.drop_cols)
 
         if self.return_df or override_return_df:
             return X

category_encoders/one_hot.py

@@ -289,7 +289,7 @@ def transform(self, X, override_return_df=False):
         X = self.get_dummies(X)
 
         if self.drop_invariant:
-            X = X.drop(columns=colself.drop_cols)
+            X = X.drop(columns=self.drop_cols)
 
         if self.return_df or override_return_df:
             return X