BB: Add type hints to MonkeyLogsDownloader.__init__()

This commit is contained in:
Mike Salvatore 2022-10-03 07:54:36 -04:00
parent 477e80bfba
commit 378e8d55ff
1 changed files with 2 additions and 1 deletions

View File

@ -4,13 +4,14 @@ from threading import Thread
from typing import List, Mapping from typing import List, Mapping
from common.types import MachineID 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 from monkey_island.cc.models import Agent, Machine
LOGGER = logging.getLogger(__name__) LOGGER = logging.getLogger(__name__)
class MonkeyLogsDownloader(object): 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.island_client = island_client
self.log_dir_path = Path(log_dir_path) self.log_dir_path = Path(log_dir_path)
self.monkey_log_paths: List[Path] = [] self.monkey_log_paths: List[Path] = []