Skip to content

Commit 9e86dd5

Browse files
committed
clean up variable naming
1 parent 96eb239 commit 9e86dd5

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

bigcode_eval/utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff 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

362362
def update_code_gens(

0 commit comments

Comments
 (0)