code: def _gen_ckpt_output(output_vars: List[str]) -> str:
      """
      Generate the return statement for checkpoint region
      """
      return f"return {', '.join(output_vars)}"
  
file_path: colossalai/colossalai/fx/codegen/activation_checkpoint_codegen.py
imports: from typing import Any, Dict, Iterable, List, Tuple
  import torch
  import colossalai
label_desc: Use image node_id 8 for calling a global function with example usage: _gen_ckpt_output(output_vars) and returns: str, str