code: def workflow_task_context(context) -> None:
      """Initialize the workflow task context.
  
      Args:
          context: The new context.
      """
      global _context
      original_context = _context
      try:
          _context = context
          yield
      finally:
          _context = original_context
  
file_path: ray/python/ray/workflow/workflow_context.py
imports: import logging
  from contextlib import contextmanager
  from dataclasses import dataclass
  from typing import Optional
  import ray
  from ray._private.ray_logging import configure_log_file, get_worker_log_file_name
  from ray.workflow.common import CheckpointModeType, WorkflowStatus
label_desc: Use image node_id 1 for calling a global function with example usage: workflow_task_context(context) without return types