@@ -197,45 +197,45 @@ def on_init(environment: Environment, **_kwargs):
197
197
test_data ["chain_id" ] = {test_data_class_name : chain_id }
198
198
if environment .parsed_options :
199
199
user_test_data .init_data (environment .parsed_options )
200
- test_data [test_data_class_name ] = user_test_data .data .to_json ()
201
- send_msg_to_workers (environment .runner , "test_data" , test_data )
202
- print ("Fetching blocks..." )
203
- if environment .parsed_options .use_latest_blocks :
204
- print (f"Using latest { user_test_data .data .size .blocks_len } blocks as test data" )
205
- logger .info (f"Using latest { user_test_data .data .size .blocks_len } blocks as test data" )
206
- for block_number in range (
207
- user_test_data .data .block_range .start , user_test_data .data .block_range .end + 1
208
- ):
209
- block = None
210
- try :
211
- block = user_test_data .fetch_block (block_number )
212
- except (BlockNotFoundError , InvalidBlockError ):
213
- pass
214
- while block is None :
200
+ test_data [test_data_class_name ] = user_test_data .data .to_json ()
201
+ send_msg_to_workers (environment .runner , "test_data" , test_data )
202
+ print ("Fetching blocks..." )
203
+ if environment .parsed_options .use_latest_blocks :
204
+ print (f"Using latest { user_test_data .data .size .blocks_len } blocks as test data" )
205
+ logger .info (f"Using latest { user_test_data .data .size .blocks_len } blocks as test data" )
206
+ for block_number in range (
207
+ user_test_data .data .block_range .start , user_test_data .data .block_range .end + 1
208
+ ):
209
+ block = None
215
210
try :
216
- block = user_test_data .fetch_latest_block ( )
211
+ block = user_test_data .fetch_block ( block_number )
217
212
except (BlockNotFoundError , InvalidBlockError ):
218
213
pass
219
- user_test_data .data .push_block (block )
220
- block_data = {test_data_class_name : block .to_json ()}
221
- send_msg_to_workers (environment .runner , "block_data" , block_data )
222
- print (user_test_data .data .stats (), end = "\r " )
214
+ while block is None :
215
+ try :
216
+ block = user_test_data .fetch_latest_block ()
217
+ except (BlockNotFoundError , InvalidBlockError ):
218
+ pass
219
+ user_test_data .data .push_block (block )
220
+ block_data = {test_data_class_name : block .to_json ()}
221
+ send_msg_to_workers (environment .runner , "block_data" , block_data )
222
+ print (user_test_data .data .stats (), end = "\r " )
223
+ else :
224
+ print (user_test_data .data .stats (), end = "\r " )
225
+ print ("\n " ) # new line after progress display upon exiting loop
223
226
else :
224
- print (user_test_data .data .stats (), end = "\r " )
225
- print ("\n " ) # new line after progress display upon exiting loop
226
- else :
227
- while user_test_data .data .size .blocks_len > len (user_test_data .data .blocks ):
228
- try :
229
- block = user_test_data .fetch_random_block (user_test_data .data .block_numbers )
230
- except (BlockNotFoundError , InvalidBlockError ):
231
- continue
232
- user_test_data .data .push_block (block )
233
- block_data = {test_data_class_name : block .to_json ()}
234
- send_msg_to_workers (environment .runner , "block_data" , block_data )
235
- print (user_test_data .data .stats (), end = "\r " )
236
- else :
237
- print (user_test_data .data .stats (), end = "\r " )
238
- print ("\n " ) # new line after progress display upon exiting loop
227
+ while user_test_data .data .size .blocks_len > len (user_test_data .data .blocks ):
228
+ try :
229
+ block = user_test_data .fetch_random_block (user_test_data .data .block_numbers )
230
+ except (BlockNotFoundError , InvalidBlockError ):
231
+ continue
232
+ user_test_data .data .push_block (block )
233
+ block_data = {test_data_class_name : block .to_json ()}
234
+ send_msg_to_workers (environment .runner , "block_data" , block_data )
235
+ print (user_test_data .data .stats (), end = "\r " )
236
+ else :
237
+ print (user_test_data .data .stats (), end = "\r " )
238
+ print ("\n " ) # new line after progress display upon exiting loop
239
239
logger .info ("Test data is ready" )
240
240
send_msg_to_workers (environment .runner , "release_lock" , {})
241
241
user_test_data .release_lock ()
0 commit comments