def store_all(data: EnsemblePredictions, datamodule: AnomalibDataModule) -> dict:
        """Store the tiled predictions in the EnsemblePredictions object."""
        tile_dict = {}
        for tile_index in [(0, 0), (0, 1), (1, 0), (1, 1)]:
            datamodule.external_collate_fn.tile_index = tile_index

            tile_prediction = []
            for batch in iter(datamodule.train_dataloader()):
                # set mock maps to just one channel of image
                batch.anomaly_maps = batch.image.clone()[:, 0, :, :].unsqueeze(1)
                # set mock pred mask to mask but add channel
                batch.pred_masks = batch.gt_mask.clone().unsqueeze(1)
                tile_prediction.append(batch)
            # save original
            tile_dict[tile_index] = copy.deepcopy(tile_prediction)
            # store to prediction storage object
            data.add_tile_prediction(tile_index, tile_prediction)

        return tile_dict
Store the tiled predictions in the EnsemblePredictions object.
tests/unit/pipelines/tiled_ensemble/test_prediction_data.py