ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 17:17:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix OS env var TRAINS_OFFLINE_MODE set to API version (e.g. 2.9)

Browse Source
This commit is contained in:
allegroai 2020-08-08 12:49:33 +03:00
parent 65003a168a
commit b710c2f19d
3 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
trains/backend_api/session/defs.py
View File

@ -1,4 +1,5 @@
from ...backend_config import EnvEntry from ...backend_config import EnvEntry
from ...backend_config.converters import safe_text_to_bool
ENV_HOST = EnvEntry("TRAINS_API_HOST", "ALG_API_HOST") ENV_HOST = EnvEntry("TRAINS_API_HOST", "ALG_API_HOST")
@ -8,4 +9,4 @@ ENV_ACCESS_KEY = EnvEntry("TRAINS_API_ACCESS_KEY", "ALG_API_ACCESS_KEY")
ENV_SECRET_KEY = EnvEntry("TRAINS_API_SECRET_KEY", "ALG_API_SECRET_KEY") ENV_SECRET_KEY = EnvEntry("TRAINS_API_SECRET_KEY", "ALG_API_SECRET_KEY")
ENV_VERBOSE = EnvEntry("TRAINS_API_VERBOSE", "ALG_API_VERBOSE", type=bool, default=False) ENV_VERBOSE = EnvEntry("TRAINS_API_VERBOSE", "ALG_API_VERBOSE", type=bool, default=False)
ENV_HOST_VERIFY_CERT = EnvEntry("TRAINS_API_HOST_VERIFY_CERT", "ALG_API_HOST_VERIFY_CERT", type=bool, default=True) ENV_HOST_VERIFY_CERT = EnvEntry("TRAINS_API_HOST_VERIFY_CERT", "ALG_API_HOST_VERIFY_CERT", type=bool, default=True)
ENV_OFFLINE_MODE = EnvEntry("TRAINS_OFFLINE_MODE", "ALG_OFFLINE_MODE", type=bool) ENV_OFFLINE_MODE = EnvEntry("TRAINS_OFFLINE_MODE", "ALG_OFFLINE_MODE", type=bool, converter=safe_text_to_bool)

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

@ -549,8 +549,9 @@ class Session(TokenManager):
if offline_api: if offline_api:
try: try:
# check cast to float, but leave original str if we pass it. # check cast to float, but leave original str if we pass it.
float(offline_api) # minimum version is 2.3
cls._offline_default_version = str(offline_api) if float(offline_api) >= 2.3:
cls._offline_default_version = str(offline_api)
except ValueError: except ValueError:
pass pass
cls.api_version = cls._offline_default_version cls.api_version = cls._offline_default_version

8
trains/backend_config/converters.py
View File

@ -24,6 +24,14 @@ def text_to_bool(value):
return bool(strtobool(value)) return bool(strtobool(value))
def safe_text_to_bool(value):
# type: (Text) -> bool
try:
return bool(strtobool(value))
except ValueError:
return bool(value)
def any_to_bool(value): def any_to_bool(value):
# type: (Optional[Union[int, float, Text]]) -> bool # type: (Optional[Union[int, float, Text]]) -> bool
if isinstance(value, six.text_type): if isinstance(value, six.text_type):