code: def load_graph(name):
      cite_graphs = ["cora", "citeseer", "pubmed"]
  
      if name in cite_graphs:
          dataset = CitationGraphDataset(name)
          graph = dataset[0]
  
          nodes = graph.nodes()
          y = graph.ndata["label"]
          train_mask = graph.ndata["train_mask"]
          val_mask = graph.ndata["test_mask"]
  
          nodes_train, y_train = nodes[train_mask], y[train_mask]
          nodes_val, y_val = nodes[val_mask], y[val_mask]
          eval_set = [(nodes_train, y_train), (nodes_val, y_val)]
  
      elif name.startswith("ogbn"):
          dataset = DglNodePropPredDataset(name)
          graph, y = dataset[0]
          split_nodes = dataset.get_idx_split()
          nodes = graph.nodes()
  
          train_idx = split_nodes["train"]
          val_idx = split_nodes["valid"]
  
          nodes_train, y_train = nodes[train_idx], y[train_idx]
          nodes_val, y_val = nodes[val_idx], y[val_idx]
          eval_set = [(nodes_train, y_train), (nodes_val, y_val)]
  
      else:
          raise ValueError("Dataset name error!")
  
      return graph, eval_set
  
imports: import argparse
  from ogb.linkproppred import *
  from ogb.nodeproppred import *
  from dgl.data import CitationGraphDataset
label_desc: Use image node_id 1 for calling a global function with example usage: load_graph(name) and returns: graph,  eval_set