fixed training_rows assignment for the alternative case of CV