fix return for complete_code

bigcode-project · Jan 4, 2024 · 88fec42 · 88fec42
1 parent 8cffbfd
commit 88fec42
Showing 1 changed file with 2 additions and 1 deletion.
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(