def _common_edge_list_args()

in graspologic/layouts/__main__.py [0:0]

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

def _common_edge_list_args(parser: argparse.ArgumentParser) -> argparse.ArgumentParser:
    parser.add_argument(
        "--edge_list",
        help="edge list in csv file. must be source,target,weight.",
        required=True,
    )
    parser.add_argument(
        "--skip_header",
        help="skip first line in csv file, corresponding to header.",
        action="store_true",
        required=False,
        default=False,
    )
    parser.add_argument(
        "--image_file",
        help="output path and filename for generated image file. "
        "required if --location_file is omitted.",
        required=False,
        default=None,
    )
    parser.add_argument(
        "--location_file",
        help="output path and filename for location file. "
        "required if --image_file is omitted.",
        required=False,
        default=None,
    )
    parser.add_argument(
        "--max_edges",
        help="maximum edges to keep during embedding. edges with low weights will be "
        "pruned to keep at most this many edges",
        type=int,
        required=False,
        default=10000000,
    )
    parser.add_argument(
        "--dpi",
        help="used with --image_file to render an image at this dpi",
        type=int,
        required=False,
        default=500,
    )
    parser.add_argument(
        "--allow_overlaps",
        help="skip the no overlap algorithm and let nodes stack as per the results of "
        "the down projection algorithm",
        action="store_true",
    )
    return parser