code: private void persistDatasetState(String datasetUrn, JobState.DatasetState datasetState)
        throws IOException {
      log.info("Persisting dataset state for dataset " + datasetUrn);
      this.jobContext.getDatasetStateStore().persistDatasetState(datasetUrn, datasetState);
    }
docs: Persist dataset state of a given dataset identified by the dataset URN.