Fix return type of web3.eth.getCode

ethereum · Mar 13, 2020 · c7b6a3c · c7b6a3c
1 parent 141275c
commit c7b6a3c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/ethpm/_utils/deployments.py b/ethpm/_utils/deployments.py
@@ -6,9 +6,6 @@
     Tuple,
 )
 
-from eth_typing import (
-    HexStr,
-)
 from eth_utils import (
     is_same_address,
     to_bytes,
@@ -17,6 +14,9 @@
 from eth_utils.toolz import (
     get_in,
 )
+from hexbytes import (
+    HexBytes,
+)
 
 from ethpm.exceptions import (
     BytecodeLinkingError,
@@ -47,7 +47,7 @@ def get_linked_deployments(deployments: Dict[str, Any]) -> Dict[str, Any]:
 
 
 def validate_linked_references(
-    link_deps: Tuple[Tuple[int, bytes], ...], bytecode: HexStr
+    link_deps: Tuple[Tuple[int, bytes], ...], bytecode: HexBytes
 ) -> None:
     """
     Validates that normalized linked_references (offset, expected_bytes)

diff --git a/web3/eth.py b/web3/eth.py
@@ -190,7 +190,7 @@ def getProof(
 
     def getCode(
         self, account: Union[Address, ChecksumAddress, ENS], block_identifier: BlockIdentifier=None
-    ) -> HexStr:
+    ) -> HexBytes:
         if block_identifier is None:
             block_identifier = self.defaultBlock
         return self.web3.manager.request_blocking(