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 329c51c commit 82f14dd
Show file tree
Hide file tree
Showing 5 changed files with 6 additions and 11 deletions.
1 change: 0 additions & 1 deletion src/aiida/brokers/broker.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
if t.TYPE_CHECKING:
from plumpy.coordinator import Coordinator

from aiida.manage.configuration.profile import Profile

__all__ = ('Broker',)

Expand Down
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
8 changes: 3 additions & 5 deletions src/aiida/brokers/rabbitmq/coordinator.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,8 @@
# -*- coding: utf-8 -*-
import concurrent.futures
from asyncio import AbstractEventLoop
import asyncio
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 @@ -71,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
1 change: 0 additions & 1 deletion src/aiida/cmdline/utils/decorators.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,6 @@
"""

import asyncio
from contextlib import contextmanager

from click_spinner import spinner
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 82f14dd

Please sign in to comment.