def parse_args()

in src/main.py [0:0]

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

def parse_args() -> dict:
    parser = argparse.ArgumentParser(
        description="GCN for proteins in DocumentDB"
    )
    parser.add_argument(
        "--batch-size",
        type=int,
        default=config.batch_size,
        help="Number of graphs (molecules) per batch",
    )
    parser.add_argument(
        "--lr", type=float, default=config.lr, help="Learning rate"
    )
    parser.add_argument(
        "--n-epochs",
        type=int,
        default=config.n_epochs,
        help="Maximum number of training epochs",
    )
    parser.add_argument(
        "--knn",
        type=int,
        default=config.knn,
        help="k used in kNN when creating protein graphs",
    )
    parser.add_argument(
        "--db-host",
        type=str,
        help="Host of DocumentDB",
    )
    parser.add_argument(
        "--db-port",
        type=str,
        help="Port of DocumentDB",
    )
    parser.add_argument(
        "--db-username",
        type=str,
        help="Username of DocumentDB",
    )
    parser.add_argument(
        "--db-password",
        type=str,
        help="Password of DocumentDB",
    )
    parser.add_argument(
        "--patience",
        type=int,
        default=config.patience,
        help="Number of epochs to wait before early stop",
    )
    return parser.parse_args().__dict__