From 21525be192aef661bf630467bf7fef1b131ff8fa Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Thu, 24 Jun 2021 12:33:48 -0400 Subject: [PATCH] agent: Use ITelem in ITelemetryMessenger.send() typehint --- .../telemetry/messengers/i_telemetry_messenger.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/infection_monkey/telemetry/messengers/i_telemetry_messenger.py b/monkey/infection_monkey/telemetry/messengers/i_telemetry_messenger.py index 7cc2efa01..cf5511e83 100644 --- a/monkey/infection_monkey/telemetry/messengers/i_telemetry_messenger.py +++ b/monkey/infection_monkey/telemetry/messengers/i_telemetry_messenger.py @@ -1,9 +1,9 @@ import abc -from infection_monkey.telemetry.base_telem import BaseTelem +from infection_monkey.telemetry.i_telem import ITelem class ITelemetryMessenger(metaclass=abc.ABCMeta): @abc.abstractmethod - def send_telemetry(self, telemetry: BaseTelem): + def send_telemetry(self, telemetry: ITelem): pass