code: def set_rank(self, rank: int, world_size: int) -> None:
      self.__rank = rank
      self.__world_size = world_size
  
file_path: ptgnn/ptgnn/baseneuralmodel/utils/data.py
imports: import random
  from typing import Callable, Iterable, Iterator, List, Optional, TypeVar
label_desc: Use image node_id 2 for calling the ShardedLazyDataIterable obj's underlying member method code with example usage: obj.set_rank(rank, world_size) without return types