code: def create_mock_pg(prefix_store, rank, world_size, timeout):
      return MockProcessGroup(rank, world_size)
  
file_path: pytorch/torch/testing/_internal/distributed/distributed_utils.py
imports: from contextlib import contextmanager
  from datetime import timedelta
  from functools import partial, wraps
  import torch.distributed
  import torch.distributed.distributed_c10d
label_desc: Use image node_id 1 for calling a global function with example usage: create_mock_pg(prefix_store, rank, world_size, timeout) and returns: MockProcessGroup