BUG: StratifiedShuffleSplit not obeying n_train

StratifiedShuffleSplit was not giving the n_train and n_test requested