code: def load_reddit(self_loop=True):
      """Load reddit dataset."""
      data = RedditDataset(self_loop=self_loop)
      g = data[0]
      g.ndata["features"] = g.ndata.pop("feat")
      g.ndata["labels"] = g.ndata.pop("label")
      return g, data.num_classes
  
file_path: dgl/examples/distributed/graphsage/partition_graph.py
imports: import argparse
  import time
  import dgl
  import torch
  from dgl.data import RedditDataset
  from ogb.nodeproppred import DglNodePropPredDataset
label_desc: Use image node_id 1 for calling a global function with example usage: load_reddit(_loop) and returns: g,  data