Experimenting with how I return results

restrictedpython_code_eval.py

@@ -433,7 +433,7 @@ def _check_correctness(check_program, timeout, task_id, completion_id,
         the results later even if execution finishes asynchronously.
     """
     manager = multiprocessing.Manager()
-    result = manager.list()
+    result = manager.dict()
 
     args = (
         check_program, result, timeout, 
@@ -449,12 +449,12 @@ def _check_correctness(check_program, timeout, task_id, completion_id,
         p.kill()
 
     if not result:
-        result.append("Result evaluates to False (probably timed out)")
+        result['result'] = "Result evaluates to False (probably timed out)"
 
     out_dict = dict(
         task_id=task_id,
-        passed=result[0] == "passed",
-        result=list(result),
+        passed=result['result'] == "passed",
+        result=dict(result),
         completion_id=completion_id,
     )
 
@@ -613,18 +613,18 @@ def _unsafe_execute(check_program, result, timeout,
                     byte_code = compile_restricted(check_program, filename="<model output>", mode="exec", policy=policy_class)
                     exec(byte_code, exec_globals, additional_locals)
             
-            result.append("passed")
+            result['result'] = "passed"
 
         except EOFError:
-            result.append("EOF error")
+            result['result'] = "EOF error"
         except TimeoutException:
-            result.append("timed out")
+            result['result'] = "timed out"
         except BaseException as e:
-            result.append(f"failed ({type(e)}): {str(e)}")
-            result.append(e)
+            result['result'] = f"failed ({type(e)}): {str(e)}"
+            result['exception'] = e
 
         if return_output and output_variable in additional_locals:  # type: ignore
-            result.append(additional_locals[output_variable])  # type: ignore
+            result['output_variable'] = additional_locals[output_variable]  # type: ignore
 
         # Needed for cleaning up.
         shutil.rmtree = rmtree