black formatting

a better solution

fix: ValueError: With n_samples=0, test_size=None and train_size=0.8, the resulting train set will be empty. Adjust any of the aforementioned parameters.

Addressing the comments.

Update cocosplit.py

Fixed issues with the coco-split utility

Add the cocosplit script