class_docstr: A batch-to-batch MapTransformFn.
class_docstr_tok: ["A","batch-to-batch","MapTransformFn","."]
code: def __init__(
      self, batch_fn: MapTransformCallable[DataBatch, DataBatch]
  ):
      self._batch_fn = batch_fn
      super().__init__(
          MapTransformFnDataType.Batch,
          MapTransformFnDataType.Batch,
      )
  
imports: import itertools
  from abc import abstractmethod
  from enum import Enum
  from typing import Any, Callable, Dict, Iterable, List, Optional, TypeVar, Union
  from ray.data._internal.block_batching.block_batching import batch_blocks
  from ray.data._internal.execution.interfaces.task_context import TaskContext
  from ray.data._internal.output_buffer import BlockOutputBuffer
  from ray.data.block import Block, BlockAccessor, DataBatch
label_desc: Use image node_id 1 to create a new BatchMapTransformFn object from inherited base classes: MapTransformFn with example: obj = BatchMapTransformFn(batch_fn)