code: def _odict_unflatten(
      values: List[Any], context: Any
  ) -> "OrderedDict[Any, Any]":
      return OrderedDict(
          (key, value) for key, value in zip(context, values)
      )
  
file_path: colossalai/colossalai/pipeline/schedule/_utils.py
imports: from collections import OrderedDict
  from typing import Any, List, Optional, Tuple
  import torch
  import torch.cuda
  from torch.nn import Module
  from torch.utils._pytree import SUPPORTED_NODES, TreeSpec, _register_pytree_node, tree_flatten, tree_map, tree_unflatten
label_desc: Use image node_id 2 for calling a global function with example usage: _odict_unflatten(values, context) and returns: OrderedDict