code: def ogbl_collate_fn(batch):
      gs, zs, xs, ws, ys = zip(*batch)
      batched_g = dgl.batch(gs)
      z = torch.cat(zs, dim=0)
      if xs[0] is not None:
          x = torch.cat(xs, dim=0)
      else:
          x = None
      if ws[0] is not None:
          edge_weights = torch.cat(ws, dim=0)
      else:
          edge_weights = None
      y = torch.tensor(ys)
  
      return batched_g, z, x, edge_weights, y
  
imports: import argparse
  import datetime
  import os
  import sys
  import time
  import dgl
  import torch
  from dgl.data.utils import load_graphs, save_graphs
  from dgl.dataloading import GraphDataLoader
  from ogb.linkproppred import DglLinkPropPredDataset, Evaluator
  from torch.nn import BCEWithLogitsLoss
  from torch.utils.data import Dataset
  from tqdm import tqdm
  from models import *
  from utils import *
label_desc: Use image node_id 1 for calling a global function with example usage: ogbl_collate_fn(batch) and returns: batched_g,  z,  x,  edge_weights,  y