def METHOD_NAME(argv):
  """Parses args for the workflow."""
  parser = argparse.ArgumentParser()
  parser.add_argument(
      '--input',
      dest='input',
      help='Path to the text file containing sentences.')
  parser.add_argument(
      '--output-dir',
      dest='output',
      required=True,
      help='Path of directory for saving trained models.')
  return parser.METHOD_NAME(argv)