def get_args()

in text-semantic-search/embeddings_extraction/run.py [0:0]

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

def get_args(argv):

  parser = argparse.ArgumentParser()

  parser.add_argument('--output_dir',
                      help='A directory location of output embeddings')

  parser.add_argument('--enable_debug',
                      action='store_true',
                      help='Enable debug options.')

  parser.add_argument('--debug_output_prefix',
                      help='Specify prefix of debug output.')

  parser.add_argument('--transform_temp_dir',
                      default='tft_temp',
                      help='A temp directory used by tf.transform.')

  parser.add_argument('--transform_export_dir',
                      default='tft_out',
                      help='A directory where tft function is saved')

  parser.add_argument('--kind',
                      help='The Cloud Datastore kind to store the items in.')

  parser.add_argument('--limit',
                      type=int,
                      default=1000000,
                      help='Maximum number of records to retrieve from BigQuery.')

  known_args, pipeline_args = parser.parse_known_args(argv)
  return known_args, pipeline_args