def get_args()

in text-semantic-search/index_builder/builder/task.py [0:0]

• 23 lines of code
• 1 McCabe index (conditional complexity)

def get_args():

  args_parser = argparse.ArgumentParser()

  args_parser.add_argument(
    '--embedding-files',
    help='GCS or local paths to embedding files',
    required=True
  )

  args_parser.add_argument(
    '--index-file',
    help='GCS or local paths to output index file',
    required=True
  )

  args_parser.add_argument(
    '--num-trees',
    help='Number of trees to build in the index',
    default=1000,
    type=int
  )

  args_parser.add_argument(
    '--job-dir',
    help='GCS or local paths to job package'
  )

  return args_parser.parse_args()