code: def teardown_process_group():
      """Ensures that the distributed process group gets closed before the next test runs."""
      yield
      if (
          torch.distributed.is_available()
          and torch.distributed.is_initialized()
      ):
          torch.distributed.destroy_process_group()
  
label_desc: Use image node_id 1 for calling a global function with example usage: teardown_process_group() without return types