ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-02-07 05:19:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add more debug printouts in k8s glue

Browse Source
This commit is contained in:
allegroai 2022-09-02 23:49:19 +03:00
parent 97c2e21dcc
commit 8b4f1eefc2
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
clearml_agent/commands/worker.py
View File

@ -1379,6 +1379,9 @@ class Worker(ServiceCommandSection):
self._session.print_configuration() self._session.print_configuration()
def resolve_daemon_queue_names(self, queues, create_if_missing=False):
return self._resolve_queue_names(queues=queues, create_if_missing=create_if_missing)
def daemon(self, queues, log_level, foreground=False, docker=False, detached=False, order_fairness=False, **kwargs): def daemon(self, queues, log_level, foreground=False, docker=False, detached=False, order_fairness=False, **kwargs):
self._apply_extra_configuration() self._apply_extra_configuration()
@ -1421,7 +1424,7 @@ class Worker(ServiceCommandSection):
# if we do not need to create queues, make sure they are valid # if we do not need to create queues, make sure they are valid
# match previous behaviour when we validated queue names before everything else # match previous behaviour when we validated queue names before everything else
queues = self._resolve_queue_names(queues, create_if_missing=kwargs.get('create_queue', False)) queues = self.resolve_daemon_queue_names(queues, create_if_missing=kwargs.get('create_queue', False))
queues_info = [ queues_info = [
q.to_dict() q.to_dict()

3
clearml_agent/glue/k8s.py
View File

@ -11,6 +11,7 @@ import subprocess
import tempfile import tempfile
from copy import deepcopy from copy import deepcopy
from pathlib import Path from pathlib import Path
from pprint import pformat
from threading import Thread from threading import Thread
from time import sleep from time import sleep
from typing import Text, List, Callable, Any, Collection, Optional, Union from typing import Text, List, Callable, Any, Collection, Optional, Union
@ -680,6 +681,8 @@ class K8sIntegration(Worker):
with open(yaml_file, 'wt') as f: with open(yaml_file, 'wt') as f:
yaml.dump(template, f) yaml.dump(template, f)
self.log.debug("Applying template:\n{}".format(pformat(template, indent=2)))
kubectl_cmd = self.KUBECTL_APPLY_CMD.format( kubectl_cmd = self.KUBECTL_APPLY_CMD.format(
task_id=task_id, task_id=task_id,
docker_image=docker_image, docker_image=docker_image,