Update data.py

should remove these line, because 'ids' in format will take care this action