Update execute.py

execute.py

@@ -40,7 +40,7 @@ def check_correctness(check_program, output, timeout, task_id, completion_id):
     manager = multiprocessing.Manager()
     result = manager.list()
 
-    p = multiprocessing.Process(target=unsafe_execute, args=(check_program, output, result, timeout))
+    p = multiprocessing.Process(target=unsafe_execute, args=(check_program, output, task_id, result, timeout))
     p.start()
     p.join(timeout=timeout + 1)
     if p.is_alive():
@@ -59,7 +59,7 @@ def check_correctness(check_program, output, timeout, task_id, completion_id):
 
 
 
-def unsafe_execute(check_program, output, result, timeout):
+def unsafe_execute(check_program, output, task_id, result, timeout):
 
     with create_tempdir():
 
@@ -84,7 +84,7 @@ def unsafe_execute(check_program, output, result, timeout):
                 exit_code = p.returncode
                 if stderrdata != '':
                     raise BaseException(stderrdata)
-                if stdoutdata.strip() == output[0].strip():            
+                if stdoutdata.strip() == output[task_id].strip():            
                     result.append("passed")
                 else:
                     result.append("not passed")