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 can_fuse(self, prev: Stage):
      context = DataContext.get_current()
      # TODO(ekl) also support fusing shuffle stages to subsequent 1:1 stages.
      if not context.optimize_fuse_shuffle_stages:
          return False
      if not self.supports_block_udf:
          return False
      if not isinstance(prev, OneToOneStage):
          return False
      if not is_task_compute(prev.compute):
          return False
      if not _are_remote_args_compatible(
          prev.ray_remote_args, self.ray_remote_args
      ):
          return False
      return True
  
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 AllToAllStage obj's underlying member method code with example usage: obj.can_fuse(prev) and returns: True, False, False, False, False, False