Update execute.py

execute.py

@@ -38,7 +38,12 @@ def check_correctness(check_program, output, timeout, task_id, completion_id):
     """
     manager = multiprocessing.Manager()
     result = manager.list()
-    unsafe_execute(check_program, output, result, timeout)
+
+    p = multiprocessing.Process(target=unsafe_execute, args=(check_program, output, result, timeout))
+    p.start()
+    p.join(timeout=timeout + 1)
+    if p.is_alive():
+        p.kill()
 
     if not result:
         result.append("timed out")