Fix when map function modifies input in-place
_The documentation is not available anymore as the PR was closed or merged._
When `function` modifies input in-place, the guarantee that columns in `remove_columns` are contained in `input` doesn't hold true anymore. Therefore we need to relax way we pop elements by checking if that column exists.