From c6849985ea6afe6cf74402dd51636f3823df3a9c Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Mon, 10 Feb 2020 10:25:57 +0200
Subject: [PATCH] Add Task.set_base_docer() and Task.get_base_docker()

---
 trains/backend_interface/task/task.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/trains/backend_interface/task/task.py b/trains/backend_interface/task/task.py
index f2e9c643..e08b264e 100644
--- a/trains/backend_interface/task/task.py
+++ b/trains/backend_interface/task/task.py
@@ -675,12 +675,24 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
     def _set_default_docker_image(self):
         if not DOCKER_IMAGE_ENV_VAR.exists():
             return
+        self.set_base_docker(DOCKER_IMAGE_ENV_VAR.get(default=""))
+
+    def set_base_docker(self, docker_cmd):
+        """
+        Set the base docker image for this experiment
+        If provided, this value will be used by trains-agent to execute this experiment
+         inside the provided docker image.
+        """
         with self._edit_lock:
             self.reload()
             execution = self.data.execution
-            execution.docker_cmd = DOCKER_IMAGE_ENV_VAR.get(default="")
+            execution.docker_cmd = docker_cmd
             self._edit(execution=execution)
 
+    def get_base_docker(self):
+        """Get the base docker command (image) set for this experiment"""
+        return self._get_task_property('execution.docker_cmd', raise_on_error=False, log_on_error=False)
+
     def set_artifacts(self, artifacts_list=None):
         """
         List of artifacts (tasks.Artifact) to update the task