code: def get_input_files(self, path, epoch=0, stage="train"):
      file_list = np.array(os.listdir(path))
      file_list.sort()
  
      if stage == "train":
          splits = np.array_split(file_list, self.__world_size)
          np.random.seed(epoch)
          np.random.shuffle(splits)
          return splits[self.rank]
      else:
          return file_list
  
imports: from .trainers_pyg import PyGTrainer
  from models.pyg import CuGraphSAGE
  import torch
  import numpy
  from torch.nn.parallel import DistributedDataParallel
  from cugraph.gnn import FeatureStore
  from cugraph_pyg.data import CuGraphStore
  from cugraph_pyg.loader import BulkSampleLoader
  import os
label_desc: Use image node_id 10 for calling the PyGCuGraphTrainer obj's underlying member method code with example usage: obj.get_input_files(path, epoch, stage) and returns: splits, file_list