Agent: Fix mypy issues in HostExploiter.py

This commit is contained in:
Kekoa Kaaikala 2022-09-21 17:47:11 +00:00
parent be4c5e7312
commit 0a9d221ad8
1 changed files with 2 additions and 1 deletions

View File

@ -7,6 +7,7 @@ from typing import Dict, Sequence
from common.event_queue import IAgentEventQueue
from common.utils.exceptions import FailedExploitationError
from infection_monkey.i_puppet import ExploiterResultData
from infection_monkey.model import VictimHost
from infection_monkey.telemetry.messengers.i_telemetry_messenger import ITelemetryMessenger
from . import IAgentBinaryRepository
@ -58,7 +59,7 @@ class HostExploiter:
# TODO: host should be VictimHost, at the moment it can't because of circular dependency
def exploit_host(
self,
host,
host: VictimHost,
servers: Sequence[str],
current_depth: int,
telemetry_messenger: ITelemetryMessenger,