From 1b7c3be65b6de765d5c3d1f03cbc94c8c8fe1f01 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Thu, 29 Sep 2022 12:43:23 -0400 Subject: [PATCH] Island: Add missing return types --- monkey/monkey_island/cc/services/initialize.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/monkey/monkey_island/cc/services/initialize.py b/monkey/monkey_island/cc/services/initialize.py index 88376ecc1..40839c388 100644 --- a/monkey/monkey_island/cc/services/initialize.py +++ b/monkey/monkey_island/cc/services/initialize.py @@ -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)