get_cost_updates() updated with batch_size