fix wait

Author: agoscinski

src/aiida/engine/processes/control.py

@@ -229,6 +229,7 @@ def _perform_actions(
     """
     futures = {}
 
+    breakpoint()
     for process in processes:
         if process.is_terminated:
             LOGGER.error(f'Process<{process.pk}> is already terminated.')
@@ -278,12 +279,10 @@ def handle_result(result):
 
     try:
         for future, process in futures.items():
-            # unwrap is need here since LoopCommunicator will also wrap a future
-            unwrapped = unwrap_kiwi_future(future)
             try:
-                result = unwrapped.result(timeout=timeout)
+                result = future.result(timeout=timeout)
             except communications.TimeoutError:
-                cancelled = unwrapped.cancel()
+                cancelled = future.cancel()
                 if cancelled:
                     LOGGER.error(f'call to {infinitive} Process<{process.pk}> timed out and was cancelled.')
                 else:
@@ -304,9 +303,10 @@ def handle_result(result):
 
         for future in concurrent.futures.as_completed(scheduled.keys(), timeout=timeout):
             process = scheduled[future]
+            unwrapped = unwrap_kiwi_future(future)
 
             try:
-                result = future.result()
+                result = unwrapped.result(timeout=timeout)
             except Exception as exception:
                 LOGGER.error(f'failed to {infinitive} Process<{process.pk}>: {exception}')
             else: