Merge pull request #120 from pyiron/future_exception

Handle Exceptions for Futures

Author: jan-janssen

pympipool/shared_functions/external_interfaces.py

@@ -249,7 +249,11 @@ def _execute_parallel_tasks_loop(interface, future_queue):
         elif "fn" in task_dict.keys() and "future" in task_dict.keys():
             f = task_dict.pop("future")
             if f.set_running_or_notify_cancel():
-                f.set_result(interface.send_and_receive_dict(input_dict=task_dict))
+                try:
+                    f.set_result(interface.send_and_receive_dict(input_dict=task_dict))
+                except Exception as thread_exeception:
+                    f.set_exception(exception=thread_exeception)
+                    raise thread_exeception
         elif "fn" in task_dict.keys() and "init" in task_dict.keys():
             interface.send_dict(input_dict=task_dict)
 

tests/test_worker.py

@@ -62,6 +62,12 @@ def test_executor_exception(self):
             with Executor(cores=1) as p:
                 p.submit(raise_error)
 
+    def test_executor_exception_future(self):
+        with self.assertRaises(RuntimeError):
+            with Executor(cores=1) as p:
+                fs = p.submit(raise_error)
+                fs.result()
+
     def test_pool_multi_core(self):
         with Executor(cores=2) as p:
             output = p.submit(mpi_funct, i=2)