Currently, the predict2() method assumes that all models used during training are also available during prediction. This causes dimension mismatch errors when some models are missing from the prediction property (for example, predicting "ChRad" after training on "BE"). Since VT hat is learned from the full set of training models, its dimensions must be aligned with the set of models present at prediction time. If models are missing, the corresponding columns of VT hat and the associated parts of the input matrix must be excluded to ensure consistency.