File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -248,7 +248,7 @@ def complete_code(
248248 # keep track of the list of generated codes
249249 # where len(code_gens) = n_tasks and len(code_gens[0]) = number of generated code samples
250250 code_gens : List [List [Optional [str ]]] = [[] for _ in range (n_tasks )]
251- intermediate_generations = [] if not intermediate_generations else intermediate_generations
251+ generations = [] if not intermediate_generations else intermediate_generations
252252 gen_token_dict = defaultdict (list ) # dict of list of generated tokens
253253 for step , batch in tqdm (
254254 enumerate (dataloader ),
@@ -335,7 +335,7 @@ def complete_code(
335335 gen_token_dict ,
336336 )
337337 with open (intermediate_save_generations_path , "w" ) as fp :
338- intermediate_save_generations = deepcopy (intermediate_generations )
338+ intermediate_save_generations = deepcopy (generations )
339339 intermediate_save_generations .extend (code_gens )
340340 json .dump (intermediate_save_generations , fp )
341341 print (
@@ -355,8 +355,8 @@ def complete_code(
355355 gen_token_dict ,
356356 )
357357
358- intermediate_generations .extend (code_gens )
359- return intermediate_generations
358+ generations .extend (code_gens )
359+ return generations
360360
361361
362362def update_code_gens (
You can’t perform that action at this time.
0 commit comments