class _MapDatasetFetcher(_BaseDataSetFetcher): def fetch(self, possibly_batched_index): if self.auto_collation: data = [self.dataset[idx] for idx in possibly_batched ...