diff --git a/src/aiida/brokers/broker.py b/src/aiida/brokers/broker.py index 941c69833d..dfac528b20 100644 --- a/src/aiida/brokers/broker.py +++ b/src/aiida/brokers/broker.py @@ -3,11 +3,11 @@ import abc import typing as t - if t.TYPE_CHECKING: - from aiida.manage.configuration.profile import Profile from plumpy.coordinator import Coordinator + from aiida.manage.configuration.profile import Profile + __all__ = ('Broker',) diff --git a/src/aiida/brokers/rabbitmq/broker.py b/src/aiida/brokers/rabbitmq/broker.py index 370afc6acf..010a7c9bee 100644 --- a/src/aiida/brokers/rabbitmq/broker.py +++ b/src/aiida/brokers/rabbitmq/broker.py @@ -15,6 +15,7 @@ if t.TYPE_CHECKING: from kiwipy.rmq import RmqThreadCommunicator + from aiida.manage.configuration.profile import Profile LOGGER = AIIDA_LOGGER.getChild('broker.rabbitmq') diff --git a/src/aiida/engine/processes/process.py b/src/aiida/engine/processes/process.py index a678b115c7..1746cee932 100644 --- a/src/aiida/engine/processes/process.py +++ b/src/aiida/engine/processes/process.py @@ -43,7 +43,6 @@ # from kiwipy.communications import UnroutableError # from plumpy.processes import ConnectionClosed # type: ignore[attr-defined] from plumpy.process_states import Finished, ProcessState - from plumpy.processes import Process as PlumpyProcess from plumpy.utils import AttributesFrozendict diff --git a/src/aiida/manage/manager.py b/src/aiida/manage/manager.py index 916589ccfb..87771b06a7 100644 --- a/src/aiida/manage/manager.py +++ b/src/aiida/manage/manager.py @@ -10,14 +10,12 @@ from __future__ import annotations +import asyncio from typing import TYPE_CHECKING, Any, Optional, Union -import asyncio -import kiwipy from plumpy.coordinator import Coordinator if TYPE_CHECKING: - from kiwipy.rmq import RmqThreadCommunicator from plumpy.process_comms import RemoteProcessThreadController from aiida.brokers.broker import Broker