added tqdm progress bar for fit and transform

Author: ragrawal

sklearn_pandas/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '1.8.0'
+__version__ = '1.8.1'
 
 from .dataframe_mapper import DataFrameMapper  # NOQA
 from .cross_validation import cross_val_score, GridSearchCV, RandomizedSearchCV  # NOQA

sklearn_pandas/dataframe_mapper.py

@@ -211,7 +211,7 @@ def fit(self, X, y=None):
         """
         self._build()
 
-        for columns, transformers, options in tqdm(self.built_features):
+        for columns, transformers, options in tqdm(self.built_features, desc='fit'):
             input_df = options.get('input_df', self.input_df)
 
             if transformers is not None:
@@ -288,9 +288,11 @@ def _transform(self, X, y=None, do_fit=False):
         if do_fit:
             self._build()
 
+
         extracted = []
         self.transformed_names_ = []
-        for columns, transformers, options in tqdm(self.built_features):
+
+        for columns, transformers, options in tqdm(self.built_features, desc='transform'):
             input_df = options.get('input_df', self.input_df)
 
             # columns could be a string or list of