code: def __init__(self, args):
      self.args = args
  
      with open("data/%s.pkl" % args.dataset_name, "rb") as f:
          graphs_info_list = pickle.load(f)
  
      self.pgl_graph_list = []
      self.graph_label_list = []
      for i in range(len(graphs_info_list) - 1):
          graph = graphs_info_list[i]
          edges_l, edges_r = graph["edge_src"], graph["edge_dst"]
  
          # add self-loops
          if self.args.dataset_name != "FRANKENSTEIN":
              num_nodes = graph["num_nodes"]
              x = np.arange(0, num_nodes)
              edges_l = np.append(edges_l, x)
              edges_r = np.append(edges_r, x)
  
          edges = list(zip(edges_l, edges_r))
          g = pgl.graph.Graph(num_nodes=graph["num_nodes"], edges=edges)
          g.node_feat["feat"] = graph["node_feat"]
          self.pgl_graph_list.append(g)
          self.graph_label_list.append(graph["label"])
  
      self.num_classes = graphs_info_list[-1]["num_classes"]
      self.num_features = graphs_info_list[-1]["num_features"]
  
imports: import sys
  import os
  import random
  import pgl
  from pgl.utils.logger import log
  from pgl.graph import Graph, MultiGraph
  import numpy
  import pickle
label_desc: Use image node_id 1 to create a new Dataset object from inherited base classes: BaseDataset with example: obj = Dataset(args)