code: def remap_keys(
      inputs: Dict[X, Any],
      mapping: Dict[X, Y],
      exclude: Optional[List[X]] = None,
  ) -> Dict[Union[X, Y], Any]:
      exclude = exclude or []
      return {
          k if k in exclude else mapping.get(k, k): v
          for k, v in inputs.items()
      }
  
file_path: pytorch_geometric/torch_geometric/sampler/utils.py
imports: from typing import Any, Dict, List, Optional, Tuple, TypeVar, Union
  import torch
  from torch import Tensor
  from torch_geometric.data import Data, HeteroData
  from torch_geometric.data.storage import EdgeStorage
  from torch_geometric.typing import EdgeType, NodeType, OptTensor
  from torch_geometric.utils import coalesce, index_sort, lexsort
  from torch_geometric.utils.sparse import index2ptr
label_desc: Use image node_id 5 for calling a global function with example usage: remap_keys(inputs, mapping, exclude) and returns: unknown