Spaces:

ZiruiWu
/

DreamOn-v0-7B

Running on Zero

root commited on Jul 18

Commit

6dd8635

1 Parent(s): 28a5503

fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -432,7 +432,7 @@ def get_example_input():
     prefix, suffix = problem['prompt'], problem['suffix']
     test_case = problem['test']
     test_case = test_case.replace('def check(candidate):', 'def run_test():')
-    test_case = test_case.replace('candidate', test_case['entry_point'])
     return prefix, '', suffix, test_case, task_id, ''
 def check_result(task_id, completion):

     prefix, suffix = problem['prompt'], problem['suffix']
     test_case = problem['test']
     test_case = test_case.replace('def check(candidate):', 'def run_test():')
+    test_case = test_case.replace('candidate', problem['entry_point'])
     return prefix, '', suffix, test_case, task_id, ''
 def check_result(task_id, completion):