class_docstr: A stage that transforms blocks holistically (e.g., shuffle).
class_docstr_tok: ["A","stage","that","transforms","blocks","holistically","(","e.g.",",","shuffle",")","."]
code: def __call__(
      self,
      blocks: BlockList,
      clear_input_blocks: bool,
      run_by_consumer: bool,
  ) -> Tuple[BlockList, dict]:
      from ray.data._internal.stage_impl import RandomizeBlocksStage
  
      in_blocks_owned_by_consumer = blocks._owned_by_consumer
      if in_blocks_owned_by_consumer:
          assert (
              run_by_consumer
          ), "Blocks owned by consumer can only be consumed by consumer"
      blocks, stage_info = self.fn(
          blocks,
          clear_input_blocks,
          self.block_udf,
          self.ray_remote_args,
      )
      assert isinstance(blocks, BlockList), blocks
  
      # RandomizeBlocksStage is an in-place transformation, so the ownership
      # of blocks doesn't change.
      if isinstance(self, RandomizeBlocksStage):
          blocks._owned_by_consumer = in_blocks_owned_by_consumer
      else:
          blocks._owned_by_consumer = run_by_consumer
  
      return blocks, stage_info
  
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 4 for calling the AllToAllStage obj's underlying member method code with example usage: obj.__call__(blocks, clear_input_blocks, run_by_consumer) and returns: blocks,  stage_info