@@ -21,7 +21,7 @@ def deploy_contracts(self, pre: Alloc, fork: Fork) -> None:
2121 code = self .generate_repeated_code (self .attack_block , self .setup , fork )
2222 self ._contract_address = pre .deploy_contract (code = code )
2323
24- def generate_transaction (self , pre : Alloc , gas_limit : int , fork : Fork ) -> Transaction :
24+ def generate_transaction (self , pre : Alloc , gas_limit : int ) -> Transaction :
2525 """Generate transaction that executes the looping contract."""
2626 if not hasattr (self , "_contract_address" ):
2727 raise ValueError ("deploy_contracts must be called before generate_transaction" )
@@ -68,7 +68,7 @@ def deploy_contracts(self, pre: Alloc, fork: Fork) -> None:
6868 caller_code = self .generate_repeated_code (code_sequence , Bytecode (), fork )
6969 self ._contract_address = pre .deploy_contract (code = caller_code )
7070
71- def generate_transaction (self , pre : Alloc , gas_limit : int , fork : Fork ) -> Transaction :
71+ def generate_transaction (self , pre : Alloc , gas_limit : int ) -> Transaction :
7272 """Generate transaction that executes the caller contract."""
7373 if not hasattr (self , "_contract_address" ):
7474 raise ValueError ("deploy_contracts must be called before generate_transaction" )
0 commit comments