Skip to content

Commit f1b6c5a

Browse files
committed
Add tagging for "incomplete generations"
1 parent 1a6fa71 commit f1b6c5a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

bigcode_eval/tasks/shadereval.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,8 @@ def remove_last_block(self, code):
195195
if not cut:
196196
if '}' in code:
197197
code = code[:code.rfind('}')] + '}'
198+
else:
199+
code = code + "// incomplete generation! \n"
198200
return code
199201

200202
def postprocess_generation(self, generation, idx):

0 commit comments

Comments
 (0)