Colab run taking too long, cant get pass split_data

Oh you’re right! I stared at it for hours and did not notice that. So silly. Thank you so much for helping.