[fix] Use training samples reurned from the model

The pylink model will return a smaller subset of training samples, if some model runs were broken off. These should then be used for the training.