@@ -50,7 +50,7 @@ def __init__(
5050provider :Optional [str ]= None ,
5151web3 :Optional [Web3 ]= None ,
5252default_slippage :float = 0.01 ,
53- poolmanager_contract_addr :Optional [AddressLike , str ]= None ,
53+ poolmanager_contract_addr :Optional [str ]= None ,
5454 )-> None :
5555"""
5656 :param address: The public address of the ETH wallet to use.
@@ -91,12 +91,14 @@ def __init__(
9191self .max_approval_check_int = int (max_approval_check_hex ,16 )
9292
9393if poolmanager_contract_addr is None :
94- self .poolmanager_contract_addr = _poolmanager_contract_addresses [self .network ]
94+ self .poolmanager_contract_addr :AddressLike = _poolmanager_contract_addresses [self .network ]
95+ else :
96+ self .poolmanager_contract_addr :AddressLike = poolmanager_contract_addr
9597
9698self .router = _load_contract (
9799self .w3 ,
98100abi_name = "uniswap-v4/poolmanager" ,
99- address = _str_to_addr (poolmanager_contract_addr ),
101+ address = _str_to_addr (self . poolmanager_contract_addr ),
100102 )
101103
102104if hasattr (self ,"poolmanager_contract" ):
@@ -155,7 +157,7 @@ def get_price(
155157try :
156158price = int (self .w3 .eth .call (signed_txn ))
157159except ContractLogicError as revert :
158- price = int (self .w3 .codec .decode (["int128[]" ,"uint160" ,"uint32" ],bytes (revert ))[1 ])
160+ price = int (self .w3 .codec .decode (["int128[]" ,"uint160" ,"uint32" ],bytes (revert . data ))[1 ])
159161return price
160162
161163def get_slot0 (