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

Fix set_base_docker() on main task to do nothing when running remotely

Browse Source
This commit is contained in:
allegroai 2020-07-30 14:34:39 +03:00
parent 65a4aa7aa9
commit e9f950f939
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
trains/task.py
View File

@ -1434,6 +1434,18 @@ class Task(_Task):
""" """
self._arguments.copy_from_dict(flatten_dictionary(dictionary)) self._arguments.copy_from_dict(flatten_dictionary(dictionary))
def set_base_docker(self, docker_cmd):
# type: (str) -> ()
"""
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.
"""
if not self.running_locally() and self.is_main_task():
return
super(Task, self).set_base_docker(docker_cmd)
def execute_remotely(self, queue_name=None, clone=False, exit_process=True): def execute_remotely(self, queue_name=None, clone=False, exit_process=True):
# type: (Optional[str], bool, bool) -> () # type: (Optional[str], bool, bool) -> ()
""" """