fix mypy errs

Author: BobTheBuidler

faster_ens/async_ens.py

@@ -515,7 +515,7 @@ async def _resolve(
 
     async def _assert_control(
         self,
-        account: ChecksumAddress,
+        account: Union[ChecksumAddress, Address],
         name: str,
         parent_owned: Optional[str] = None,
     ) -> None:
@@ -570,7 +570,7 @@ async def _claim_ownership(
     async def _setup_reverse(
         self,
         name: Optional[str],
-        address: ChecksumAddress,
+        address: Union[ChecksumAddress, Address],
         transact: Optional["TxParams"] = None,
     ) -> HexBytes:
         name = normalize_name(name) if name else ""

faster_ens/ens.py

@@ -158,7 +158,7 @@ def address(
         if coin_type is None:
             # don't validate `addr(bytes32)` interface id since extended resolvers
             # can implement a "resolve" function as of ENSIP-10
-            return cast(ChecksumAddress, self._resolve(name, "addr"))
+            return cast(Optional[ChecksumAddress], self._resolve(name, "addr"))
         else:
             r = self.resolver(name)
             _validate_resolver_and_interface_id(
@@ -244,7 +244,7 @@ def name(self, address: ChecksumAddress) -> Optional[str]:
     def setup_name(
         self,
         name: str,
-        address: Optional[ChecksumAddress] = None,
+        address: Optional[Union[ChecksumAddress, Address]] = None,
         transact: Optional["TxParams"] = None,
     ) -> HexBytes:
         """
@@ -496,7 +496,7 @@ def _resolve(
 
     def _assert_control(
         self,
-        account: ChecksumAddress,
+        account: Union[ChecksumAddress, Address],
         name: str,
         parent_owned: Optional[str] = None,
     ) -> None:
@@ -547,7 +547,7 @@ def _claim_ownership(
     def _setup_reverse(
         self,
         name: Optional[str],
-        address: ChecksumAddress,
+        address: Union[ChecksumAddress, Address],
         transact: Optional["TxParams"] = None,
     ) -> HexBytes:
         name = normalize_name(name) if name else ""

faster_ens/utils.py

@@ -266,7 +266,8 @@ def raw_name_to_hash(name: str) -> HexBytes:
 
 
 def address_in(
-    address: ChecksumAddress, addresses: Collection[ChecksumAddress]
+    address: Union[ChecksumAddress, Address],
+    addresses: Collection[Union[ChecksumAddress, Address]],
 ) -> bool:
     return any(is_same_address(address, item) for item in addresses)