[IterableDataset] Fix `drop_last_batch`in map after shuffling or sharding
It was not taken into account e.g. when passing to a DataLoader with num_workers>0

Fix https://github.com/huggingface/datasets/issues/6565