Moved X.columns check into base encoder

Author: nercisla

category_encoders/target_encoder.py

@@ -165,8 +165,6 @@ def _check_dict_key_tuples(self, d):
 
     def _fit(self, X, y, **kwargs):
         if isinstance(self.hierarchy, dict):
-            if not set(self.cols).issubset(X.columns):
-                raise ValueError('X does not contain the columns listed in cols')
             X_hier = pd.DataFrame()
             for switch in self.hierarchy:
                 if switch in self.cols:

category_encoders/utils.py

@@ -295,6 +295,9 @@ def fit(self, X, y=None, **kwargs):
         self._dim = X.shape[1]
         self._get_fit_columns(X)
 
+        if not set(self.cols).issubset(X.columns):
+            raise ValueError('X does not contain the columns listed in cols')
+
         if self.handle_missing == 'error':
             if X[self.cols].isnull().any().any():
                 raise ValueError('Columns to be encoded can not contain null')