rebase amend

aiidateam · Dec 30, 2024 · 92cbdef · 92cbdef
1 parent 8347f29
commit 92cbdef
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/plumpy/rmq/communications.py b/src/plumpy/rmq/communications.py
@@ -9,7 +9,8 @@
 
 import kiwipy
 
-from plumpy import futures
+from plumpy.futures import create_task
+from plumpy.rmq.futures import wrap_to_concurrent_future
 from plumpy.utils import ensure_coroutine
 
 __all__ = [
@@ -72,7 +73,7 @@ def converted(communicator: kiwipy.Communicator, *args: Any, **kwargs: Any) -> k
             return kiwi_future
 
         msg_fn = functools.partial(coro, communicator, *args, **kwargs)
-        task_future = futures.create_task(msg_fn, loop)
+        task_future = create_task(msg_fn, loop)
         return wrap_to_concurrent_future(task_future)
 
     return converted