tiny tweak to `train_test_split()` #135