code: def parse_args():
      """
      Helper function parsing the command line options
      """
      parser = ArgumentParser(
          description="PyTorch distributed training launch "
          "helper utilty that will spawn up "
          "parties for MPC scripts"
      )
  
      # Optional arguments for the launch helper
      parser.add_argument(
          "--world_size",
          type=int,
          default=1,
          help="The number of parties to launch."
          "Each party acts as its own process",
      )
  
      # positional
      parser.add_argument(
          "training_script",
          type=str,
          help="The full path to the single GPU training "
          "program/script to be launched in parallel, "
          "followed by all the arguments for the "
          "training script",
      )
  
      # rest from the training program
      parser.add_argument("training_script_args", nargs=REMAINDER)
      return parser.parse_args()
  
imports: import os
  import subprocess
  import uuid
  from argparse import ArgumentParser, REMAINDER
label_desc: Use image node_id 2 for calling a global function with example usage: parse_args() and returns: parser