ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 18:52:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Avoid retries when verifying invalid credentials

Browse Source
This commit is contained in:
allegroai 2020-01-06 17:05:37 +02:00
parent 54ae340ccb
commit 8585d7e134
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
trains/backend_api/session/session.py
View File

@ -85,6 +85,7 @@ class Session(TokenManager):
initialize_logging=True, initialize_logging=True,
client=None, client=None,
config=None, config=None,
http_retries_config=None,
**kwargs **kwargs
): ):
@ -127,9 +128,8 @@ class Session(TokenManager):
raise ValueError("host is required in init or config") raise ValueError("host is required in init or config")
self.__host = host.strip("/") self.__host = host.strip("/")
http_retries_config = self.config.get( http_retries_config = http_retries_config or self.config.get(
"api.http.retries", ConfigTree() "api.http.retries", ConfigTree()).as_plain_ordered_dict()
).as_plain_ordered_dict()
http_retries_config["status_forcelist"] = self._retry_codes http_retries_config["status_forcelist"] = self._retry_codes
self.__http_session = get_http_session_with_retry(**http_retries_config) self.__http_session = get_http_session_with_retry(**http_retries_config)

9
trains/config/default/__main__.py
View File

@ -177,15 +177,16 @@ def verify_credentials(api_host, credentials):
try: try:
print('Verifying credentials ...') print('Verifying credentials ...')
if api_host: if api_host:
Session(api_key=credentials['access_key'], secret_key=credentials['secret_key'], host=api_host) Session(api_key=credentials['access_key'], secret_key=credentials['secret_key'], host=api_host,
http_retries_config={"total": 1})
print('Credentials verified!') print('Credentials verified!')
return True return True
else: else:
print("Can't verify credentials") print("Can't verify credentials")
return False return False
except Exception: except Exception as ex:
print('Error: could not verify credentials: key={} secret={}'.format( print('Exception: {}\nError: could not verify credentials: key={} secret={}'.format(
credentials.get('access_key'), credentials.get('secret_key'))) str(ex), credentials.get('access_key'), credentials.get('secret_key')))
return False return False