code: def get_halo_excnahger_1d(self):
      peer_ranks = [i for i in range(self.world_size)]
      pool = PeerMemoryPool(64 * 1024, 2 * 1024 * 1024, peer_ranks)
      halo_exchanger_1d = PeerHaloExchanger1d(
          peer_ranks, self.rank, pool, TestPeerMemory.HALF_HALO
      )
      return halo_exchanger_1d
  
file_path: apex/apex/contrib/test/peer_memory/test_peer_halo_exchange_module.py
imports: import unittest
  import torch
  from torch.testing._internal import common_utils
  from apex.transformer.testing.distributed_test_base import NcclDistributedTestBase
label_desc: Use image node_id 3 for calling the TestPeerMemory obj's underlying member method code with example usage: obj.get_halo_excnahger_1d() and returns: halo_exchanger_1d