code: def map_keys(batch: Batch) -> Batch:
      """DMVR dataset returns 'image' and 'label'. We want 'inputs' and 'label'."""
  
      batch["inputs"] = batch["image"]
      return batch
  
imports: import functools
  import os
  from typing import Dict, Iterator, List, Optional, Text, Tuple, Union
  from absl import logging
  from dmvr import builders
  from dmvr import modalities
  from dmvr import video_dataset
  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
  from scenic.dataset_lib import video_ops
  from scenic.projects.vivit.data import file_utils
  import tensorflow
label_desc: Use image node_id 3 for calling a global function with example usage: map_keys(batch) and returns: batch