From 21f01292f7ec7722e11b51bdf20964189f1add51 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Wed, 28 Sep 2022 17:47:04 +0530 Subject: [PATCH] Agent: Fix type hint in IIslandAPIClient.send_log() --- .../infection_monkey/island_api_client/i_island_api_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/infection_monkey/island_api_client/i_island_api_client.py b/monkey/infection_monkey/island_api_client/i_island_api_client.py index d14c8deeb..86d42ee88 100644 --- a/monkey/infection_monkey/island_api_client/i_island_api_client.py +++ b/monkey/infection_monkey/island_api_client/i_island_api_client.py @@ -5,7 +5,7 @@ from common import AgentRegistrationData, AgentSignals, OperatingSystem from common.agent_configuration import AgentConfiguration from common.agent_events import AbstractAgentEvent from common.credentials import Credentials -from common.types import SocketAddress +from common.types import AgentID, SocketAddress class IIslandAPIClient(ABC): @@ -30,7 +30,7 @@ class IIslandAPIClient(ABC): """ @abstractmethod - def send_log(self, agent_id: int, log_contents: str): + def send_log(self, agent_id: AgentID, log_contents: str): """ Send the contents of the agent's log to the island