Fixed. This logic is now part of `_safe_split` in `cross_validation.py`.