Add `drop_last_batch` to `IterableDataset.map`
_The documentation is not available anymore as the PR was closed or merged._
Addresses this comment: https://github.com/huggingface/datasets/pull/3801#pullrequestreview-901736921