Re-apply input columns change
_The documentation is not available anymore as the PR was closed or merged._
Fixes the `filter` + `input_columns` combination, which is used in the `transformers` examples for instance.

Revert #5006 (which in turn reverts #4971)

Fix https://github.com/huggingface/datasets/issues/4858