From d4ec506fc42a449f5e528a23f824e26ac45a538a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Niels=20M=C3=BCndler?= Date: Tue, 20 Feb 2024 20:35:44 +0100 Subject: [PATCH] Fix crucial references to withdrawal (#316) --- pycardano/txbuilder.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/pycardano/txbuilder.py b/pycardano/txbuilder.py index 85bbe763..ab27a780 100644 --- a/pycardano/txbuilder.py +++ b/pycardano/txbuilder.py @@ -313,12 +313,12 @@ def add_withdrawal_script( TransactionBuilder: Current transaction builder. """ if redeemer: - if redeemer.tag is not None and redeemer.tag != RedeemerTag.REWARD: + if redeemer.tag is not None and redeemer.tag != RedeemerTag.WITHDRAWAL: raise InvalidArgumentException( - f"Expect the redeemer tag's type to be {RedeemerTag.REWARD}, " + f"Expect the redeemer tag's type to be {RedeemerTag.WITHDRAWAL}, " f"but got {redeemer.tag} instead." ) - redeemer.tag = RedeemerTag.REWARD + redeemer.tag = RedeemerTag.WITHDRAWAL self._consolidate_redeemer(redeemer) if isinstance(script, UTxO): @@ -1259,9 +1259,7 @@ def _update_execution_units( assert ( r.tag is not None ), "Expected tag of redeemer to be set, but found None" - tagname = ( - r.tag.name.lower() if r.tag != RedeemerTag.REWARD else "withdrawal" - ) + tagname = r.tag.name.lower() key = f"{tagname}:{r.index}" if ( key not in estimated_execution_units