Revert input_columns change
_The documentation is not available anymore as the PR was closed or merged._ Merging this one and I'll check if it fixes the `transformers` CI before doing a patch release
Revert https://github.com/huggingface/datasets/pull/4971

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