[MRG] StratifiedShuffleSplit bug

Fixes #2764