From 8a4666fba22444abdebc8de34495463833531f7c Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Mon, 19 Sep 2022 17:24:29 +0530 Subject: [PATCH] Agent: Fix Island API client import in network/relay/utils.py --- monkey/infection_monkey/network/relay/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/monkey/infection_monkey/network/relay/utils.py b/monkey/infection_monkey/network/relay/utils.py index 2d2bcc3d7..875d49649 100644 --- a/monkey/infection_monkey/network/relay/utils.py +++ b/monkey/infection_monkey/network/relay/utils.py @@ -6,8 +6,8 @@ from typing import Dict, Iterable, Iterator, MutableMapping, Optional from common.network.network_utils import address_to_ip_port from infection_monkey.network.relay import RELAY_CONTROL_MESSAGE_REMOVE_FROM_WAITLIST -from infection_monkey.transport import IslandApiClient -from infection_monkey.transport.island_api_client_errors import ( +from infection_monkey.transport import ( + IslandAPIClient, IslandAPIConnectionError, IslandAPIError, IslandAPITimeoutError, @@ -54,7 +54,7 @@ def _check_if_island_server(server: str) -> bool: logger.debug(f"Trying to connect to server: {server}") try: - _ = IslandApiClient(server) + _ = IslandAPIClient(server) return True except IslandAPIConnectionError as err: