Transformer to prevent column leakage of test

prevent_column_leakage_transformer.py

class comprehensive_features_custom(BaseEstimator, TransformerMixin):
  
   def __init__(self, comprehensive_features):
       self.comprehensive_features = comprehensive_features
 
   def fit(self, X, y = None):
       return self
 
   def transform(self, X):
       X_post = X.reindex(labels=comprehensive_features, axis=1)
       return X_post
 
set_comprehensive_features = comprehensive_features_custom(comprehensive_features)
 
#comprehensive preprocessing pipe
comprehensive_preprocessing = Pipeline([
   ('combined_preprocessing',combined_preprocessing),
   ('set_comprehensive_features',set_comprehensive_features)
])