Skip to content

Commit 4008fb7

Browse files
authored
MAINT clarifications in ColumnTransformer._update_transformers (scikit-learn#11323)
1 parent 02fad7a commit 4008fb7

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

sklearn/compose/_column_transformer.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -290,22 +290,25 @@ def get_feature_names(self):
290290
return feature_names
291291

292292
def _update_fitted_transformers(self, transformers):
293+
# transformers are fitted; excludes 'drop' cases
293294
transformers = iter(transformers)
294295
transformers_ = []
295296

296297
for name, old, column in self.transformers:
297298
if old == 'drop':
298-
trans = old
299+
trans = 'drop'
299300
elif old == 'passthrough':
300301
# FunctionTransformer is present in list of transformers,
301302
# so get next transformer, but save original string
302303
next(transformers)
303-
trans = old
304+
trans = 'passthrough'
304305
else:
305306
trans = next(transformers)
306307

307308
transformers_.append((name, trans, column))
308309

310+
# sanity check that transformers is exhausted
311+
assert not list(transformers)
309312
self.transformers_ = transformers_
310313

311314
def _validate_output(self, result):

0 commit comments

Comments
 (0)