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

Add pass TRAINS_DOCKER_IMAGE into docker for interactive sessions

Browse Source
This commit is contained in:
allegroai 2020-11-29 23:16:40 +02:00
parent 0cf485f7a9
commit 74b646af9e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trains_agent/commands/worker.py
View File

@ -2362,6 +2362,8 @@ class Worker(ServiceCommandSection):
raise ValueError('Error: could not copy .ssh directory into: {}'.format(new_ssh_cache))
base_cmd += ['-e', 'TRAINS_WORKER_ID='+worker_id, ]
# update the docker image, so the system knows where it runs
base_cmd += ['-e', 'TRAINS_DOCKER_IMAGE={} {}'.format(docker_image, ' '.join(docker_arguments)).strip()]
# if we are running a RC version, install the same version in the docker
# because the default latest, will be a release version (not RC)