ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-02-07 13:26:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix FORCE_LOCAL_CLEARML_AGENT_WHEEL when running from a Windows host

Browse Source
This commit is contained in:
allegroai 2022-02-09 18:28:17 +02:00
parent b1ee3e105b
commit 14c48d0a78
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
clearml_agent/commands/worker.py
View File

@ -21,6 +21,7 @@ from distutils.spawn import find_executable
from distutils.util import strtobool from distutils.util import strtobool
from functools import partial from functools import partial
from itertools import chain from itertools import chain
from os.path import basename
from tempfile import mkdtemp, NamedTemporaryFile from tempfile import mkdtemp, NamedTemporaryFile
from time import sleep, time from time import sleep, time
from typing import Text, Optional, Any, Tuple, List from typing import Text, Optional, Any, Tuple, List
@ -3383,7 +3384,7 @@ class Worker(ServiceCommandSection):
agent_install_bash_script = [] agent_install_bash_script = []
if os.environ.get('FORCE_LOCAL_CLEARML_AGENT_WHEEL'): if os.environ.get('FORCE_LOCAL_CLEARML_AGENT_WHEEL'):
local_wheel = os.path.expanduser(os.environ.get('FORCE_LOCAL_CLEARML_AGENT_WHEEL')) local_wheel = os.path.expanduser(os.environ.get('FORCE_LOCAL_CLEARML_AGENT_WHEEL'))
docker_wheel = str(Path('/tmp') / Path(local_wheel).name) docker_wheel = '/tmp/{}'.format(basename(local_wheel))
base_cmd += ['-v', local_wheel + ':' + docker_wheel] base_cmd += ['-v', local_wheel + ':' + docker_wheel]
clearml_agent_wheel = '\"{}\"'.format(docker_wheel) clearml_agent_wheel = '\"{}\"'.format(docker_wheel)
elif os.environ.get('FORCE_CLEARML_AGENT_REPO'): elif os.environ.get('FORCE_CLEARML_AGENT_REPO'):