My split_data function runs multiple times

[balaji.ambresh]

Notebook markdown on the expected output is correct.
Please fix your code. Here’s a thread that lists the steps.