Skip to content

Commit

Permalink
Fix crucial references to withdrawal (Python-Cardano#316)
Browse files Browse the repository at this point in the history
  • Loading branch information
nielstron authored Feb 20, 2024
1 parent 997d3d2 commit d4ec506
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions pycardano/txbuilder.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit d4ec506

Please sign in to comment.