@@ -21,7 +21,7 @@ def deploy_contracts(self, pre: Alloc, fork: Fork) -> None:
21
21
code = self .generate_repeated_code (self .attack_block , self .setup , fork )
22
22
self ._contract_address = pre .deploy_contract (code = code )
23
23
24
- def generate_transaction (self , pre : Alloc , gas_limit : int , fork : Fork ) -> Transaction :
24
+ def generate_transaction (self , pre : Alloc , gas_limit : int ) -> Transaction :
25
25
"""Generate transaction that executes the looping contract."""
26
26
if not hasattr (self , "_contract_address" ):
27
27
raise ValueError ("deploy_contracts must be called before generate_transaction" )
@@ -68,7 +68,7 @@ def deploy_contracts(self, pre: Alloc, fork: Fork) -> None:
68
68
caller_code = self .generate_repeated_code (code_sequence , Bytecode (), fork )
69
69
self ._contract_address = pre .deploy_contract (code = caller_code )
70
70
71
- def generate_transaction (self , pre : Alloc , gas_limit : int , fork : Fork ) -> Transaction :
71
+ def generate_transaction (self , pre : Alloc , gas_limit : int ) -> Transaction :
72
72
"""Generate transaction that executes the caller contract."""
73
73
if not hasattr (self , "_contract_address" ):
74
74
raise ValueError ("deploy_contracts must be called before generate_transaction" )
0 commit comments