From 74b646af9e21ca1fe117645051b40ad19f42bfad Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Sun, 29 Nov 2020 23:16:40 +0200
Subject: [PATCH] Add pass TRAINS_DOCKER_IMAGE into docker for interactive
 sessions

---
 trains_agent/commands/worker.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/trains_agent/commands/worker.py b/trains_agent/commands/worker.py
index 8fcf1a4..b382c0f 100644
--- a/trains_agent/commands/worker.py
+++ b/trains_agent/commands/worker.py
@@ -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)