Align remove columns behavior and input dict mutation in `map` with previous behavior
Align the `remove_columns` behavior and input dict mutation in `map` with the behavior before https://github.com/huggingface/datasets/pull/5252.