Align remove columns behavior and input dict mutation in `map` with previous behavior
_The documentation is not available anymore as the PR was closed or merged._
Align the `remove_columns` behavior and input dict mutation in `map` with the behavior before https://github.com/huggingface/datasets/pull/5252.