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/_analyzer/fx/codegen.py
imports: from typing import Any, Dict, List, Tuple
  import torch
  from torch.fx.graph import PythonCode, _custom_builtins, _format_target, _is_from_torch, _Namespace, _origin_type_map, _register_custom_builtin, inplace_methods, magic_methods
  from torch.fx.node import Argument, Node, _get_qualified_name, _type_repr, map_arg
  import colossalai
  from colossalai.fx._compatibility import compatibility
label_desc: Use image node_id 2 for calling a global function with example usage: _gen_ckpt_output(output_vars) and returns: str, str