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 k8s set task container if using default image/arguments

Browse Source
This commit is contained in:
allegroai 2021-07-15 10:26:09 +03:00
parent 4c9410c5fe
commit 6470b16b70
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
clearml_agent/glue/k8s.py
View File

@ -17,7 +17,7 @@ from typing import Text, List, Callable, Any, Collection, Optional, Union
import yaml import yaml
from clearml_agent.commands.events import Events from clearml_agent.commands.events import Events
from clearml_agent.commands.worker import Worker, get_task_container from clearml_agent.commands.worker import Worker, get_task_container, set_task_container
from clearml_agent.definitions import ENV_DOCKER_IMAGE from clearml_agent.definitions import ENV_DOCKER_IMAGE
from clearml_agent.errors import APIError from clearml_agent.errors import APIError
from clearml_agent.helper.base import safe_remove_file from clearml_agent.helper.base import safe_remove_file
@ -281,6 +281,9 @@ class K8sIntegration(Worker):
ENV_DOCKER_IMAGE.get() or self._session.config.get("agent.default_docker.image", "nvidia/cuda") ENV_DOCKER_IMAGE.get() or self._session.config.get("agent.default_docker.image", "nvidia/cuda")
) )
container['arguments'] = self._session.config.get("agent.default_docker.arguments", None) container['arguments'] = self._session.config.get("agent.default_docker.arguments", None)
set_task_container(
self._session, task_id, docker_image=container['image'], docker_arguments=container['arguments']
)
# get the clearml.conf encoded file # get the clearml.conf encoded file
# noinspection PyProtectedMember # noinspection PyProtectedMember