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

Question

I want to create pandas udf function for Pyspark in which I am using train_test_split function and returns a dataframe

This is what I have done.