From 7facf302a401032210f737b368a5359f7ee6bd20 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Tue, 15 Mar 2022 13:13:07 -0400 Subject: [PATCH] Agent: Rename unused '_' parameter to architecture in get_agent_binary --- monkey/common/utils/exceptions.py | 4 ---- monkey/infection_monkey/exploit/caching_agent_repository.py | 2 +- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/monkey/common/utils/exceptions.py b/monkey/common/utils/exceptions.py index 2a0e369e9..945e35415 100644 --- a/monkey/common/utils/exceptions.py +++ b/monkey/common/utils/exceptions.py @@ -18,10 +18,6 @@ class IncorrectCredentialsError(Exception): """ Raise to indicate that authentication failed """ -class InvalidAWSKeys(Exception): - """ Raise to indicate that AWS API keys are invalid""" - - class NoInternetError(Exception): """ Raise to indicate problems caused when no internet connection is present""" diff --git a/monkey/infection_monkey/exploit/caching_agent_repository.py b/monkey/infection_monkey/exploit/caching_agent_repository.py index 8a55b3f63..9d24746ad 100644 --- a/monkey/infection_monkey/exploit/caching_agent_repository.py +++ b/monkey/infection_monkey/exploit/caching_agent_repository.py @@ -22,7 +22,7 @@ class CachingAgentRepository(IAgentRepository): self._proxies = proxies self._lock = threading.Lock() - def get_agent_binary(self, os: str, _: str = None) -> io.BytesIO: + def get_agent_binary(self, os: str, architecture: str = None) -> io.BytesIO: # If multiple calls to get_agent_binary() are made simultaneously before the result of # _download_binary_from_island() is cached, then multiple requests will be sent to the # island. Add a mutex in front of the call to _download_agent_binary_from_island() so