fix --predictions with missing test set