code: def add_number_map_fn(
      dataset: tf.data.Dataset,
      increment: int,
      input_name: str,
      output_name: str,
  ) -> tf.data.Dataset:
      def f(ex):
          ex[output_name] = ex[input_name] + increment
          return ex
  
      return dataset.map(f)
  
imports: import functools
  import tensorflow
  import tensorflow_datasets
  from tensorflow_datasets.core.dataset_builders import view_builder
label_desc: Use image node_id 3 for calling a global function with example usage: add_number_map_fn(dataset, increment, input_name, output_name) and returns: dataset, ex