Merge pull request #47 from blocknative/fix/case-no-abi

Fix parsing when no abi is present
blocknative · Jul 2, 2022 · 9fb3d52 · 9fb3d52
2 parents b750e05 + 3e7e778
commit 9fb3d52
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/blocknative/__init__.py b/blocknative/__init__.py
@@ -1 +1 @@
-__version__ = '0.2.7'
+__version__ = '0.2.8'
diff --git a/blocknative/stream.py b/blocknative/stream.py
@@ -125,7 +125,7 @@ def subscribe_address(
         address: str,
         callback: Callback,
         filters: List[dict] = None,
-        abi: List[dict] = None,
+        abi: Union[List[dict], str] = None,
     ):
         """Subscribes to an address to listen to any incoming and
         outgoing transactions that occur on that address.
@@ -140,10 +140,13 @@ def subscribe_address(
         if self.blockchain == BN_ETHEREUM:
             address = address.lower()
 
+        if isinstance(abi, str):
+            abi = json.loads(abi)
+
         # Add this subscription to the registry
         self._subscription_registry[address] = Subscription(
             callback,
-            {"filters": filters, "abi": json.loads(abi)},
+            {"filters": filters, "abi": abi},
             SubscriptionType.ADDRESS,
         )