def main()

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

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

def main():
    parser = argparse.ArgumentParser(description="Rank embeddings for object_name_or_event_ID, event_type and object_type")

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

    get_ranged_completions(args.dataset, args.entity_name, args.indices_path)