Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Dec 27, 2024
1 parent 03f7a5b commit ca1dd09
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 8 deletions.
3 changes: 1 addition & 2 deletions src/aiida/brokers/rabbitmq/broker.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,8 @@
import functools
import typing as t

from plumpy.rmq import RemoteProcessThreadController, RmqCoordinator
from plumpy import ProcessController
from plumpy.rmq.process_control import RemoteProcessController
from plumpy.rmq import RemoteProcessThreadController

from aiida.brokers.broker import Broker
from aiida.brokers.rabbitmq.coordinator import RmqLoopCoordinator
Expand Down
7 changes: 3 additions & 4 deletions src/aiida/brokers/rabbitmq/coordinator.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,8 @@
# -*- coding: utf-8 -*-
import concurrent.futures
from asyncio import AbstractEventLoop
from typing import Generic, TypeVar, final
import kiwipy
import concurrent.futures

import kiwipy
from plumpy.exceptions import CoordinatorConnectionError
from plumpy.rmq.communications import convert_to_comm

Expand Down Expand Up @@ -70,7 +69,7 @@ def broadcast_send(
subject=None,
correlation_id=None,
):
from aio_pika.exceptions import ChannelInvalidStateError, AMQPConnectionError
from aio_pika.exceptions import AMQPConnectionError, ChannelInvalidStateError

try:
rsp = self._comm.broadcast_send(body, sender, subject, correlation_id)
Expand Down
4 changes: 2 additions & 2 deletions src/aiida/engine/runners.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,12 +22,12 @@
from plumpy.coordinator import Coordinator
from plumpy.events import reset_event_loop_policy, set_event_loop_policy
from plumpy.persistence import Persister
from plumpy.rmq import RemoteProcessThreadController, wrap_communicator
from plumpy.rmq import RemoteProcessThreadController

from aiida.brokers import Broker
from aiida.common import exceptions
from aiida.orm import ProcessNode, load_node
from aiida.plugins.utils import PluginVersionProvider
from aiida.brokers import Broker

from . import transports, utils
from .processes import Process, ProcessBuilder, ProcessState, futures
Expand Down

0 comments on commit ca1dd09

Please sign in to comment.