Island: Add missing return types

This commit is contained in:
Mike Salvatore 2022-09-29 12:43:23 -04:00
parent 82e08ba157
commit 1b7c3be65b
1 changed files with 6 additions and 2 deletions

View File

@ -122,11 +122,15 @@ def _register_event_queues(container: DIContainer):
container.register_instance(IIslandEventQueue, decorated_island_event_queue)
def _decorate_agent_event_queue(agent_event_queue: IAgentEventQueue, lock: threading.Lock):
def _decorate_agent_event_queue(
agent_event_queue: IAgentEventQueue, lock: threading.Lock
) -> IAgentEventQueue:
return LockingAgentEventQueueDecorator(agent_event_queue, lock)
def _decorate_island_event_queue(island_event_queue: IIslandEventQueue, lock: threading.Lock):
def _decorate_island_event_queue(
island_event_queue: IIslandEventQueue, lock: threading.Lock
) -> IIslandEventQueue:
return LockingIslandEventQueueDecorator(island_event_queue, lock)