class_docstr: Represents a Dataset transform stage (e.g., map or shuffle).
class_docstr_tok: ["Represents","a","Dataset","transform","stage","(","e.g.",",","map","or","shuffle",")","."]
docstr: Execute this stage against the given blocks.
docstr_tok: ["Execute","this","stage","against","the","given","blocks","."]
code: def __call__(
      self, blocks: BlockList, clear_input_blocks: bool
  ) -> Tuple[BlockList, dict]:
      """Execute this stage against the given blocks."""
      raise NotImplementedError
  
imports: import copy
  import itertools
  from typing import TYPE_CHECKING, Any, Callable, Dict, Iterable, Iterator, List, Optional, Tuple, Union
  import ray
  from ray._private.internal_api import get_memory_info_reply, get_state_from_address
  from ray.data._internal.block_list import BlockList
  from ray.data._internal.compute import ActorPoolStrategy, BlockTransform, CallableClass, ComputeStrategy, TaskPoolStrategy, UserDefinedFunction, get_compute, is_task_compute
  from ray.data._internal.dataset_logger import DatasetLogger
  from ray.data._internal.execution.interfaces import TaskContext
  from ray.data._internal.lazy_block_list import LazyBlockList
  from ray.data._internal.logical.interfaces.logical_operator import LogicalOperator
  from ray.data._internal.logical.operators.from_operators import AbstractFrom
  from ray.data._internal.logical.operators.input_data_operator import InputData
  from ray.data._internal.logical.operators.read_operator import Read
  from ray.data._internal.logical.rules.operator_fusion import _are_remote_args_compatible
  from ray.data._internal.stats import DatasetStats, DatasetStatsSummary
  from ray.data._internal.util import create_dataset_tag, unify_block_metadata_schema
  from ray.data.block import Block, BlockMetadata
  from ray.data.context import DataContext
  from ray.types import ObjectRef
  from ray.util.debug import log_once
label_desc: Use image node_id 2 for calling the Stage obj's underlying member method code with example usage: obj.__call__(blocks, clear_input_blocks) without return types