From f5279a6cb108143ab921afc40585303465dec66c Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
Date: Mon, 19 Sep 2022 17:09:52 +0530
Subject: [PATCH] Agent: Import Island API errors in transport/__init__.py

---
 monkey/infection_monkey/transport/__init__.py          | 6 ++++++
 monkey/infection_monkey/transport/island_api_client.py | 2 +-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/monkey/infection_monkey/transport/__init__.py b/monkey/infection_monkey/transport/__init__.py
index 5b5cccf83..c9b793a50 100644
--- a/monkey/infection_monkey/transport/__init__.py
+++ b/monkey/infection_monkey/transport/__init__.py
@@ -1,2 +1,8 @@
 from infection_monkey.transport.http import LockedHTTPServer
+from .island_api_client_errors import (
+    IslandAPIConnectionError,
+    IslandAPIError,
+    IslandAPIRequestFailedError,
+    IslandAPITimeoutError,
+)
 from .island_api_client import IslandAPIClient
diff --git a/monkey/infection_monkey/transport/island_api_client.py b/monkey/infection_monkey/transport/island_api_client.py
index 56bb72383..f872de295 100644
--- a/monkey/infection_monkey/transport/island_api_client.py
+++ b/monkey/infection_monkey/transport/island_api_client.py
@@ -3,7 +3,7 @@ import logging
 import requests
 
 from common.common_consts.timeouts import MEDIUM_REQUEST_TIMEOUT
-from infection_monkey.transport.island_api_client_errors import (
+from infection_monkey.transport import (
     IslandAPIConnectionError,
     IslandAPIError,
     IslandAPITimeoutError,