def METHOD_NAME(benchmark_spec):
  """Create metadata dict to be used in run results.
  Args:
    benchmark_spec: The benchmark specification. Contains all data that is
      required to run the benchmark.
  Returns:
    metadata dict
  """
  metadata = mnist_benchmark.CreateMetadataDict(benchmark_spec)
  metadata.update({
      'model': benchmark_spec.model,
      'problem': benchmark_spec.problem,
      'hparams_set': benchmark_spec.hparams_set,
      'data_dir': benchmark_spec.data_dir,
      'model_dir': benchmark_spec.model_dir,
      'train_steps': benchmark_spec.train_steps,
      'eval_steps': benchmark_spec.eval_steps})
  return metadata