diff --git a/src/plumpy/rmq/communications.py b/src/plumpy/rmq/communications.py index 61f89fc1..b27b65a1 100644 --- 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