code: def __iter__(self):
      if self.shuffle:
          np.random.shuffle(self.subgs)
  
      for subg in self.subgs:
          yield subg
  
file_path: PGL/ogb_examples/nodeproppred/ogbn-proteins/unimp_cross_feat/partition.py
imports: import numpy
  import math
  from pgl.sample import extract_edges_from_nodes
label_desc: Use image node_id 2 for calling the RandomPartition obj's underlying member method code with example usage: obj.__iter__() without return types