From 378e8d55ff778509b99125c7984aa6ee2b9b8f0c Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Mon, 3 Oct 2022 07:54:36 -0400 Subject: [PATCH] BB: Add type hints to MonkeyLogsDownloader.__init__() --- .../monkey_zoo/blackbox/log_handlers/monkey_logs_downloader.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/envs/monkey_zoo/blackbox/log_handlers/monkey_logs_downloader.py b/envs/monkey_zoo/blackbox/log_handlers/monkey_logs_downloader.py index 6f1815e2f..26ff851ed 100644 --- a/envs/monkey_zoo/blackbox/log_handlers/monkey_logs_downloader.py +++ b/envs/monkey_zoo/blackbox/log_handlers/monkey_logs_downloader.py @@ -4,13 +4,14 @@ from threading import Thread from typing import List, Mapping from common.types import MachineID +from envs.monkey_zoo.blackbox.island_client.monkey_island_client import MonkeyIslandClient from monkey_island.cc.models import Agent, Machine LOGGER = logging.getLogger(__name__) class MonkeyLogsDownloader(object): - def __init__(self, island_client, log_dir_path): + def __init__(self, island_client: MonkeyIslandClient, log_dir_path: str): self.island_client = island_client self.log_dir_path = Path(log_dir_path) self.monkey_log_paths: List[Path] = []