code: def collate_fn(data_list: List[Data]) -> Batch:
      batch = Batch()
      for key in data_list[0].keys():
          batch[key] = default_collate(
              [data[key] for data in data_list]
          )
      return batch
  
imports: from typing import List, Union
  import torch
  from torch.utils.data.dataloader import default_collate
  from torch_geometric.data import Batch, Data, Dataset
label_desc: Use image node_id 1 for calling a global function with example usage: collate_fn(data_list) and returns: batch