diff --git a/src/plumpy/broadcast_filter.py b/src/plumpy/broadcast_filter.py index 61b27095..6ec6c41e 100644 --- a/src/plumpy/broadcast_filter.py +++ b/src/plumpy/broadcast_filter.py @@ -1,4 +1,5 @@ # -*- coding: utf-8 -*- +# type: ignore import re import typing diff --git a/src/plumpy/controller.py b/src/plumpy/controller.py index 9f2793a0..55fe4da6 100644 --- a/src/plumpy/controller.py +++ b/src/plumpy/controller.py @@ -52,8 +52,7 @@ def play_process(self, pid: 'PID_TYPE') -> ProcessResult: ... def play_all(self) -> None: - """Play all processes that are subscribed to the same coordinator - """ + """Play all processes that are subscribed to the same coordinator""" def kill_process(self, pid: 'PID_TYPE', msg_text: str | None = None) -> Any: """Kill the process diff --git a/src/plumpy/coordinator.py b/src/plumpy/coordinator.py index e647961e..fb330700 100644 --- a/src/plumpy/coordinator.py +++ b/src/plumpy/coordinator.py @@ -1,8 +1,8 @@ # -*- coding: utf-8 -*- from __future__ import annotations -from typing import TYPE_CHECKING, Any, Callable, Hashable, Protocol from re import Pattern +from typing import TYPE_CHECKING, Any, Callable, Hashable, Protocol if TYPE_CHECKING: ID_TYPE = Hashable @@ -36,7 +36,11 @@ def unhook_broadcast_receiver(self, identifier: 'ID_TYPE | None') -> None: ... def unhook_task_receiver(self, identifier: 'ID_TYPE') -> None: ... - def rpc_send(self, recipient_id: Hashable, msg: Any,) -> Any: ... + def rpc_send( + self, + recipient_id: Hashable, + msg: Any, + ) -> Any: ... def broadcast_send( self, diff --git a/src/plumpy/message.py b/src/plumpy/message.py index 098277e1..99d215ff 100644 --- a/src/plumpy/message.py +++ b/src/plumpy/message.py @@ -7,7 +7,6 @@ import logging from typing import TYPE_CHECKING, Any, Dict, Optional, Sequence, Union, cast -from plumpy.coordinator import Coordinator from plumpy.exceptions import PersistenceError, TaskRejectedError from . import loaders, persistence diff --git a/src/plumpy/processes.py b/src/plumpy/processes.py index 75737574..55d09e26 100644 --- a/src/plumpy/processes.py +++ b/src/plumpy/processes.py @@ -34,9 +34,7 @@ cast, ) -import kiwipy - -from plumpy.broadcast_filter import BroadcastFilter +from plumpy.broadcast_filter import BroadcastFilter # type: ignore from plumpy.coordinator import Coordinator try: diff --git a/src/plumpy/rmq/communications.py b/src/plumpy/rmq/communications.py index 3ed3877a..e9f227b0 100644 --- a/src/plumpy/rmq/communications.py +++ b/src/plumpy/rmq/communications.py @@ -82,7 +82,9 @@ def converted(communicator: kiwipy.Communicator, *args: Any, **kwargs: Any) -> k CommT = TypeVar('CommT', bound=kiwipy.Communicator) -def wrap_communicator(communicator: CommT, loop: Optional[asyncio.AbstractEventLoop] = None) -> 'LoopCommunicator[CommT]': +def wrap_communicator( + communicator: CommT, loop: Optional[asyncio.AbstractEventLoop] = None +) -> 'LoopCommunicator[CommT]': """ Wrap a communicator such that all callbacks made to any subscribers are scheduled on the given event loop. diff --git a/tests/utils.py b/tests/utils.py index c0cf0f52..fe8f9900 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -1,5 +1,6 @@ # -*- coding: utf-8 -*- """Utilities for tests""" +from __future__ import annotations import asyncio import collections @@ -105,7 +106,7 @@ def hook_broadcast_receiver( self._broadcast_receivers[identifier] = receiver return identifier - def unhook_broadcast_receiver(self, identifier: 'ID_TYPE | None') -> None: + def unhook_broadcast_receiver(self, identifier: 'ID_TYPE | None') -> None: self._ensure_open() try: del self._broadcast_receivers[identifier]