def METHOD_NAME(self) -> "DataLoader":
    self.labelled._predict_input = self.filter_unlabelled_data(self._dataset.pool)
    dataloader = self.labelled._predict_dataloader()
    dataloader.collate_fn = create_worker_input_transform_processor(
        RunningStage.TRAINING, self.labelled.input_transform
    )
    return dataloader