kiwipy.Future -> concurrent.futures.Future

src/plumpy/processes.py

@@ -932,7 +932,7 @@ def message_receive(self, _comm: Communicator, msg: Dict[str, Any]) -> Any:
 
     def broadcast_receive(
         self, _comm: Communicator, body: Any, sender: Any, subject: Any, correlation_id: Any
-    ) -> Optional[kiwipy.Future]:
+    ) -> Optional[concurrent.futures.Future]:
         """
         Coroutine called when the process receives a message from the communicator
 
@@ -962,7 +962,7 @@ def broadcast_receive(
 
         return fn
 
-    def _schedule_rpc(self, callback: Callable[..., Any], *args: Any, **kwargs: Any) -> kiwipy.Future:
+    def _schedule_rpc(self, callback: Callable[..., Any], *args: Any, **kwargs: Any) -> concurrent.futures.Future:
         """
         Schedule a call to a callback as a result of an RPC communication call, this will return
         a future that resolves to the final result (even after one or more layer of futures being
@@ -977,7 +977,7 @@ def _schedule_rpc(self, callback: Callable[..., Any], *args: Any, **kwargs: Any)
         :return: a kiwi future that resolves to the outcome of the callback
 
         """
-        kiwi_future = kiwipy.Future()
+        kiwi_future = concurrent.futures.Future()
 
         async def run_callback() -> None:
             with capture_exceptions(kiwi_future):