ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-06-26 18:16:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve docker host-mount support, use TRAINS_AGENT_DOCKER_HOST_MOUNT env var

Browse Source
This commit is contained in:
allegroai
2020-05-09 20:02:46 +03:00
parent 7c87797a40
commit 53f511f536
3 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
trains_agent/helper/singleton.py
View File

@@ -4,7 +4,7 @@ from time import sleep
from glob import glob
from tempfile import gettempdir, NamedTemporaryFile
from trains_agent.definitions import ENV_K8S_HOST_MOUNT
from trains_agent.definitions import ENV_DOCKER_HOST_MOUNT
from trains_agent.helper.base import warning
@@ -85,7 +85,7 @@ class Singleton(object):
pass
worker = None
if api_client and os.environ.get(ENV_K8S_HOST_MOUNT) and uid:
if api_client and ENV_DOCKER_HOST_MOUNT.get() and uid:
try:
worker = [w for w in api_client.workers.get_all() if w.id == uid]
except Exception:
@@ -134,8 +134,8 @@ class Singleton(object):
@classmethod
def _get_temp_folder(cls):
if os.environ.get(ENV_K8S_HOST_MOUNT):
return os.environ.get(ENV_K8S_HOST_MOUNT).split(':')[-1]
if ENV_DOCKER_HOST_MOUNT.get():
return ENV_DOCKER_HOST_MOUNT.get().split(':')[-1]
return gettempdir()
@classmethod