code: def load_ogb_dataset(dataset):
      """
      Load OGB dataset
      Args:
          dataset(str): name of dataset (ogbl-collab, ogbl-ddi, ogbl-citation)
  
      Returns:
          graph(DGLGraph): graph
          split_edge(dict): split edge
  
      """
      dataset = DglLinkPropPredDataset(name=dataset)
      split_edge = dataset.get_edge_split()
      graph = dataset[0]
  
      return graph, split_edge
  
file_path: dgl/examples/pytorch/seal/utils.py
imports: import argparse
  import dgl
  import numpy
  import pandas
  import torch
  from ogb.linkproppred import DglLinkPropPredDataset, Evaluator
  from scipy.sparse.csgraph import shortest_path
label_desc: Use image node_id 2 for calling a global function with example usage: load_ogb_dataset(dataset) and returns: graph,  split_edge