def main()

in src/scripts/extract_data.py [0:0]

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

def main():
    parser = argparse.ArgumentParser(description="Extract entities for given dataset messages.")

    parser.add_argument("-d", "--dataset", required=True, help="Dataset name", type=Dataset, choices=list(Dataset))
    parser.add_argument("-i", "--indices_path", type=str, help="Path to file with indices", default=None)
    args = parser.parse_args()

    extract_data(args.dataset, args.indices_path)