def get_batch_tiles(self, batch_index: int) -> dict[tuple[int, int], ImageBatch]:
        """Get all tiles of current batch from class dictionary.

        Called by merging mechanism.

        Args:
            batch_index (int): Index of current batch of tiles to be returned.

        Returns:
            dict[tuple[int, int], dict]: Dictionary mapping tile index to predicted data, for provided batch index.
        """
        batch_data = {}

        for index, batches in self.all_data.items():
            batch_data[index] = batches[batch_index]

        return batch_data
Get all tiles of current batch from class dictionary.

        Called by merging mechanism.

        Args:
            batch_index (int): Index of current batch of tiles to be returned.

        Returns:
            dict[tuple[int, int], dict]: Dictionary mapping tile index to predicted data, for provided batch index.
        
src/anomalib/pipelines/tiled_ensemble/components/utils/prediction_data.py