Add __repr__

tradingstrategy-ai · Jun 19, 2023 · 77ea507 · 77ea507
1 parent 1099ef2
commit 77ea507
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/eth_defi/enzyme/events.py b/eth_defi/enzyme/events.py
@@ -28,7 +28,6 @@
 logger = logging.getLogger(__name__)
 
 
-@dataclass(slots=True)
 class EnzymeBalanceEvent:
     """Enzyme deposit/redeem event wrapper.
 
@@ -168,7 +167,6 @@ def receiver(self) -> HexAddress:
         return convert_uint256_bytes_to_address(HexBytes(self.event_data["topics"][1]))
 
 
-@dataclass(slots=True)
 class Redemption(EnzymeBalanceEvent):
     """Enzyme deposit event wrapper.
 
@@ -259,6 +257,9 @@ class LiveBalance:
     #:
     balance: Decimal
 
+    def __repr__(self):
+        return f"<Token {self.token}, balance {self.balance}>"
+
 
 def fetch_vault_balance_events(
     vault: Vault,