diff --git a/bigcode_eval/tasks/shadereval.py b/bigcode_eval/tasks/shadereval.py index 7e76cf196..3e5790b02 100644 --- a/bigcode_eval/tasks/shadereval.py +++ b/bigcode_eval/tasks/shadereval.py @@ -195,6 +195,8 @@ def remove_last_block(self, code): if not cut: if '}' in code: code = code[:code.rfind('}')] + '}' + else: + code = code + "// incomplete generation! \n" return code def postprocess_generation(self, generation, idx):