reset the batch size to 1000 for import_dataset