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