fix return for complete_code

maxmatical · maxmatical · commit 88fec422a3c2 · 2024-01-04T10:42:09.000-05:00
diff --git a/bigcode_eval/utils.py b/bigcode_eval/utils.py
@@ -355,7 +355,8 @@ def complete_code(
         gen_token_dict,
     )
 
-    return intermediate_generations.extend(code_gens)
+    intermediate_generations.extend(code_gens)
+    return intermediate_generations
 
 
 def update_code_gens(

Original file line number	Diff line number	Diff line change
`@@ -355,7 +355,8 @@ def complete_code(`
`355`	`355`	`gen_token_dict,`
`356`	`356`	`)`
`357`	`357`
`358`		`- return intermediate_generations.extend(code_gens)`
	`358`	`+ intermediate_generations.extend(code_gens)`
	`359`	`+ return intermediate_generations`
`359`	`360`
`360`	`361`
`361`	`362`	`def update_code_gens(`