code: def collate_fn(data_list):
      return data_list
  
imports: from typing import List, Union
  import torch
  from torch_geometric.data import Dataset
  from torch_geometric.data.data import BaseData
label_desc: Use image node_id 1 for calling a global function with example usage: collate_fn(data_list) and returns: data_list