code: def load_dgl_graph():
      from ogb.nodeproppred import DglNodePropPredDataset
  
      dataset = DglNodePropPredDataset(
          name="ogbn-mag", root="/datasets/vjawa/gnn/"
      )
      split_idx = dataset.get_idx_split()
      g, labels = dataset[0]
      # Uncomment for obgn-mag
      labels = labels["paper"].flatten()
      # labels = labels
      # transform = Compose([ToSimple(), AddReverse()])
      # g = transform(g)
      return g, labels, dataset.num_classes, split_idx
  
imports: import dgl
  import torch
  import torch.nn.functional
  import time
  import argparse
  from cugraph_dgl import c
label_desc: Use image node_id 1 for calling a global function with example usage: load_dgl_graph() and returns: g,  labels,  dataset,  split_idx