Understanding and Resolving the TypeError: Singleton Array Cannot Be Considered a Valid Collection Using scikit-learn's `train_test_split` Function