fixed bug in split_data.py