def get_args()

in retail/recommendation-system/bqml-scann/embeddings_exporter/runner.py [0:0]

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

def get_args(argv):

  args_parser = argparse.ArgumentParser()

  args_parser.add_argument('--bq_dataset_name',
                           help='BigQuery dataset name.',
                           required=True)

  args_parser.add_argument('--embeddings_table_name',
                           help='BigQuery table name where the embeddings are stored.',
                           required=True)

  args_parser.add_argument('--output_dir',
                           help='GCS location where the embedding CSV files will be stored.',
                           required=True)

  return args_parser.parse_known_args()