From 416cb30577690b7341885c46ee6413c6f83458c1 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Wed, 10 Aug 2022 07:29:13 -0400 Subject: [PATCH] Common: Add type hint ty PyPubSubEventQueue constructor --- monkey/common/event_queue/pypubsub_event_queue.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/monkey/common/event_queue/pypubsub_event_queue.py b/monkey/common/event_queue/pypubsub_event_queue.py index 882857050..efac70881 100644 --- a/monkey/common/event_queue/pypubsub_event_queue.py +++ b/monkey/common/event_queue/pypubsub_event_queue.py @@ -1,6 +1,7 @@ from typing import Callable from common.events import AbstractEvent +from pubsub.core import Publisher from .i_event_queue import IEventQueue @@ -8,7 +9,7 @@ INTERNAL_ALL_EVENT_TYPES_TOPIC = "internal_all_event_types" class PyPubSubEventQueue(IEventQueue): - def __init__(self, pypubsub_publisher): + def __init__(self, pypubsub_publisher: Publisher): self._pypubsub_publisher = pypubsub_publisher def subscribe_all(self, subscriber: Callable[[AbstractEvent], None]):