def get_args()

in retail/recommendation-system/bqml-scann/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-path',
    help='GCS or local paths to embedding files',
    required=True
  )

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

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

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

  return args_parser.parse_args()