Week 1 Assignment cant run

The first assignment for the week I cant get passed the split_data test cell. It takes way too long to run. I had to leave it running while I slept last night but unfortunately the runtime disconnected before it could finish the cell. Currently, it’s been running for half an hour and it’s still on the training set. How are you guys doing this assignment?

Please click my name and message your notebook as an attachment.

Please see this thread. Do remember that python is space sensitive when correcting the mistake.