Add third arg state override set. Fix ethereum#1921

edd34 · Apr 1, 2021 · 16d3971 · 16d3971
1 parent 836996f
commit 16d3971
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/web3/eth.py b/web3/eth.py
@@ -521,8 +521,9 @@ def sign_munger(
     def call_munger(
         self,
         transaction: TxParams,
-        block_identifier: Optional[BlockIdentifier] = None
-    ) -> Tuple[TxParams, BlockIdentifier]:
+        block_identifier: Optional[BlockIdentifier] = None,
+        state_override: Optional[TxParams] = None,
+    ) -> Tuple[TxParams, BlockIdentifier, TxParams]:
         # TODO: move to middleware
         if 'from' not in transaction and is_checksum_address(self.default_account):
             transaction = assoc(transaction, 'from', self.default_account)
@@ -531,7 +532,7 @@ def call_munger(
         if block_identifier is None:
             block_identifier = self.default_block
 
-        return (transaction, block_identifier)
+        return (transaction, block_identifier, state_override)
 
     call: Method[Callable[..., Union[bytes, bytearray]]] = Method(
         RPC.eth_call,