else deleted from train_test_split because it was not needed

[MohammadJavadRamezanpour]

since your if and elif are raising an exception, the flowing else was not necessary, I deleted it to make the code achieve the same goal with fewer lines