Skip to content

Commit ca1dd09

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 03f7a5b commit ca1dd09

File tree

3 files changed

+6
-8
lines changed

3 files changed

+6
-8
lines changed

src/aiida/brokers/rabbitmq/broker.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,8 @@
66
import functools
77
import typing as t
88

9-
from plumpy.rmq import RemoteProcessThreadController, RmqCoordinator
109
from plumpy import ProcessController
11-
from plumpy.rmq.process_control import RemoteProcessController
10+
from plumpy.rmq import RemoteProcessThreadController
1211

1312
from aiida.brokers.broker import Broker
1413
from aiida.brokers.rabbitmq.coordinator import RmqLoopCoordinator

src/aiida/brokers/rabbitmq/coordinator.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
1-
# -*- coding: utf-8 -*-
1+
import concurrent.futures
22
from asyncio import AbstractEventLoop
33
from typing import Generic, TypeVar, final
4-
import kiwipy
5-
import concurrent.futures
64

5+
import kiwipy
76
from plumpy.exceptions import CoordinatorConnectionError
87
from plumpy.rmq.communications import convert_to_comm
98

@@ -70,7 +69,7 @@ def broadcast_send(
7069
subject=None,
7170
correlation_id=None,
7271
):
73-
from aio_pika.exceptions import ChannelInvalidStateError, AMQPConnectionError
72+
from aio_pika.exceptions import AMQPConnectionError, ChannelInvalidStateError
7473

7574
try:
7675
rsp = self._comm.broadcast_send(body, sender, subject, correlation_id)

src/aiida/engine/runners.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,12 @@
2222
from plumpy.coordinator import Coordinator
2323
from plumpy.events import reset_event_loop_policy, set_event_loop_policy
2424
from plumpy.persistence import Persister
25-
from plumpy.rmq import RemoteProcessThreadController, wrap_communicator
25+
from plumpy.rmq import RemoteProcessThreadController
2626

27+
from aiida.brokers import Broker
2728
from aiida.common import exceptions
2829
from aiida.orm import ProcessNode, load_node
2930
from aiida.plugins.utils import PluginVersionProvider
30-
from aiida.brokers import Broker
3131

3232
from . import transports, utils
3333
from .processes import Process, ProcessBuilder, ProcessState, futures

0 commit comments

Comments
 (0)