code: private WorkUnitStream prepareWorkUnits(WorkUnitStream workUnits, JobState jobState) {
      return workUnits.transform(new WorkUnitPreparator(this.jobContext.getJobId()));
    }
docs: Prepare the flattened {@link WorkUnit}s for execution by populating the job and task IDs.