From c6c63521d877488cb26b37e1abbb0a05008453bf Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sat, 21 Dec 2024 11:10:15 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/aiida/brokers/broker.py | 4 ++-- src/aiida/brokers/rabbitmq/broker.py | 1 + src/aiida/engine/processes/process.py | 1 - src/aiida/manage/manager.py | 4 +--- 4 files changed, 4 insertions(+), 6 deletions(-) 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