code: def load_dataset(dataset_name):
      multilabel = False
      if dataset_name in [
          "reddit",
          "cora",
          "citeseer",
          "pubmed",
          "yelp",
          "flickr",
      ]:
          g, n_classes = load_dgl(dataset_name)
          multilabel = dataset_name in ["yelp"]
          if multilabel:
              g.ndata["labels"] = g.ndata["labels"].to(dtype=th.float32)
      elif dataset_name in [
          "ogbn-products",
          "ogbn-arxiv",
          "ogbn-papers100M",
          "ogbn-mag240M",
      ]:
          g, n_classes = load_ogb(dataset_name)
      else:
          raise ValueError("unknown dataset")
  
      return g, n_classes, multilabel
  
file_path: dgl/examples/pytorch/labor/load_graph.py
imports: import dgl
  import torch
label_desc: Use image node_id 6 for calling a global function with example usage: load_dataset(dataset_name) and returns: g,  n_classes,  multilabel