Improve endOffsets for TaskMetadata by updating immediately after polling a new batch Reviewers: Anna Sophie Blee-Goldman <ableegoldman@apache.org>