code: def map_keys(batch: Batch) -> Batch:
      """DMVR dataset returns 'image' and 'label'. We want 'inputs' and 'label'."""
      batch["inputs"] = batch.pop(builders.IMAGE_FEATURE_NAME)
      return batch
  
imports: import functools
  from typing import Dict, Iterator, Optional, Text, Tuple, Union
  from absl import logging
  from dmvr import builders
  from dmvr import modalities
  from flax import jax_utils
  import jax
  import jax.numpy
  import ml_collections
  import numpy
  from scenic.dataset_lib import dataset_utils
  from scenic.dataset_lib import datasets
  import scenic.projects.objectvivit.dataset_utils
  from scenic.projects.vivit.data.video_tfrecord_dataset import TFRecordDatasetFactory
  import tensorflow
label_desc: Use image node_id 3 for calling a global function with example usage: map_keys(batch) and returns: batch