def METHOD_NAME(proof: Literal["correct", "incorrect"]) -> bytes:
    """
    Format depending on whether we want a correct proof or not.
    """
    kzg_commitment = INF_POINT
    kzg_proof = INF_POINT
    z = Z
    # INF_POINT commitment and proof evaluate to 0 on all z values
    y = 0 if proof == "correct" else 1
    versioned_hash = Spec.kzg_to_versioned_hash(kzg_commitment)
    return (
        versioned_hash
        + z.to_bytes(32, "little")
        + y.to_bytes(32, "little")
        + kzg_commitment
        + kzg_proof
    )