test_split.py script to do split into train and test Usage: python train_test_split.py data.csv instructions.json