code: def parse_args():
      """Parse command line arguments."""
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--push",
          action="store_true",
          help="Push image after creation. This requires that you enter a tag that includes a registry via --tag",
      )
      parser.add_argument(
          "--tag",
          default="wandb-launch-agent",
          help="Tag for the image",
      )
      parser.add_argument(
          "--platform",
          default="linux/amd64",
          help="Platform to use, e.g. 'linux/amd64'",
      )
      return parser.parse_args()
  
imports: import argparse
  import os
  from wandb.docker import build, push
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: parser