code: private static DatasetRepository getDatasetRepository(JobContext jobContext) {
      Configuration conf = Hadoop.JobContext.getConfiguration.invoke(jobContext);
      DatasetRepository repo = DatasetRepositories.repositoryFor(conf.get(KITE_OUTPUT_URI));
      if (repo instanceof TemporaryDatasetRepositoryAccessor) {
        Dataset<Object> dataset = load(jobContext).getDataset();
        String namespace = dataset.getNamespace();
        repo = ((TemporaryDatasetRepositoryAccessor) repo)
            .getTemporaryRepository(namespace, getJobDatasetName(jobContext));
      }
      return repo;
    }
docs: TODO: Remove the need to use DatasetRepositories.repositoryFor()