Skip to content

Commit 1a8ea06

Browse files
committed
adding column name as description
1 parent d845c11 commit 1a8ea06

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

sklearn_pandas/dataframe_mapper.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -210,8 +210,9 @@ def fit(self, X, y=None):
210210
211211
"""
212212
self._build()
213-
214-
for columns, transformers, options in tqdm(self.built_features, desc='fit'):
213+
pbar = tqdm(self.built_features)
214+
for columns, transformers, options in pbar:
215+
pbar.set_description("[Fit] %s" % columns)
215216
input_df = options.get('input_df', self.input_df)
216217

217218
if transformers is not None:
@@ -291,8 +292,9 @@ def _transform(self, X, y=None, do_fit=False):
291292

292293
extracted = []
293294
self.transformed_names_ = []
294-
295-
for columns, transformers, options in tqdm(self.built_features, desc='transform'):
295+
pbar = tqdm(self.built_features)
296+
for columns, transformers, options in pbar:
297+
pbar.set_description("[Transform] %s" % columns)
296298
input_df = options.get('input_df', self.input_df)
297299

298300
# columns could be a string or list of

0 commit comments

Comments
 (0)