title: Fix when map function modifies input in-place
comments: _The documentation is not available anymore as the PR was closed or merged._
body: 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.