File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff 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 ):
You can’t perform that action at this time.
0 commit comments