code: def graph_saint_randomwalk_sampler(graph, batch_nodes, max_depth=3):
      subgraph = graph_saint_random_walk_sample(
          graph, batch_nodes, max_depth
      )
      sub_node_index = subgraph.reindex_from_parrent_nodes(batch_nodes)
      return subgraph, sub_node_index
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from __future__ import unicode_literals
  from __future__ import absolute_import
  from dataloader.base_dataloader import BaseDataGenerator
  from utils.to_undirected import to_undirected
  import ssl
  from pgl.contrib.ogb.nodeproppred.dataset_pgl import PglNodePropPredDataset
  from pgl.sample import graph_saint_random_walk_sample
  from ogb.nodeproppred import Evaluator
  import tqdm
  from collections import namedtuple
  import pgl
  import numpy
  import copy
label_desc: Use image node_id 4 for calling a global function with example usage: graph_saint_randomwalk_sampler(graph, batch_nodes, max_depth) and returns: subgraph,  sub_node_index