We often need to plot aspects of a model for a particular feature - eg partial dependence, feature importance, etc. This requires keeping track of the features through the pipeline. The current way we do this can break if the features change as a component of the pipeline - either because some features are filtered out or a more general feature transform such as PCA is applied.