StratifiedShuffleSplit bug

Fixes #2764