Inference and Item's labels

When doing inference (i.e. load_learner), is there a way to get predictions while maintaining the labels of the data? get_preds on test data doesn’t keep the item’s label. I know you should need this inference, but if I want to extract embeddings from the step, I need the label of the embedding.

You can at all times maintain a dictionary or list of (x, labels).