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

Allow worker_id override using environment variable

Browse Source
This commit is contained in:
allegroai 2020-05-22 10:20:43 +03:00
parent 3491646ea9
commit 63ca84a84f
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -145,7 +145,7 @@ class Session(TokenManager):
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)
self.__worker = worker or gethostname() self.__worker = worker or self.get_worker_host_name()
self.__max_req_size = self.config.get("api.http.max_req_size", None) self.__max_req_size = self.config.get("api.http.max_req_size", None)
if not self.__max_req_size: if not self.__max_req_size:
@ -530,6 +530,11 @@ class Session(TokenManager):
return version_tuple(cls.api_version) >= version_tuple(str(min_api_version)) return version_tuple(cls.api_version) >= version_tuple(str(min_api_version))
@classmethod
def get_worker_host_name(cls):
from ...config import dev_worker_name
return dev_worker_name() or gethostname()
def _do_refresh_token(self, old_token, exp=None): def _do_refresh_token(self, old_token, exp=None):
""" TokenManager abstract method implementation. """ TokenManager abstract method implementation.
Here we ignore the old token and simply obtain a new token. Here we ignore the old token and simply obtain a new token.