reduced batch size to avoid DeadlineError