code: def __init__(self, data_index, data_label, mode="train"):
      if (
          int(paddle.distributed.get_world_size()) == 1
          or mode != "train"
      ):
          self.data = [data_index, data_label]
      else:
          self.data = [
              data_index[
                  int(paddle.distributed.get_rank()) :: int(
                      paddle.distributed.get_world_size()
                  )
              ],
              data_label[
                  int(paddle.distributed.get_rank()) :: int(
                      paddle.distributed.get_world_size()
                  )
              ],
          ]
  
file_path: PGL/examples/graphsage/cpu_sample_version/dataset.py
imports: import os
  import paddle
  import numpy
  from pgl import graph_kernel
  from pgl.utils.logger import log
  from pgl.sampling import graphsage_sample
  from pgl.utils.data import Dataset
label_desc: Use image node_id 1 to create a new ShardedDataset object from inherited base classes: Dataset with example: obj = ShardedDataset(data_index, data_label, mode)