code: def color_debug(text, prefix=" ", color="blue"):
      color = color.upper()
      print(getattr(Back, color), prefix, Style.RESET_ALL, text)
  
imports: import argparse
  import os
  import warnings
  import torch
  import torch.distributed.rpc
  import torch.multiprocessing
  from torch import nn
  from torch._C._distributed_rpc import _is_current_rpc_agent_set
  from colossalai.legacy import launch
  from colossalai.legacy.pipeline.pipeline_process_group import ppg
  from colossalai.logging import disable_existing_loggers
label_desc: Use image node_id 1 for calling a global function with example usage: color_debug(text, prefix, color) without return types