From 43443ccf08234cbd89bce791bbdab469db0a0e19 Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Mon, 1 Apr 2024 11:37:01 +0300 Subject: [PATCH] Pass task_id when resolving k8s template --- clearml_agent/glue/k8s.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/clearml_agent/glue/k8s.py b/clearml_agent/glue/k8s.py index 3a5cb1f..6620077 100644 --- a/clearml_agent/glue/k8s.py +++ b/clearml_agent/glue/k8s.py @@ -570,7 +570,7 @@ class K8sIntegration(Worker): print("Kubernetes scheduling task id={}".format(task_id)) try: - template = self._resolve_template(task_session, task_data, queue) + template = self._resolve_template(task_session, task_data, queue, task_id) except Exception as ex: print("ERROR: Failed resolving template (skipping): {}".format(ex)) return @@ -1106,7 +1106,7 @@ class K8sIntegration(Worker): self._session, queue=queue, get_task_info=get_task_info ) - def _resolve_template(self, task_session, task_data, queue): + def _resolve_template(self, task_session, task_data, queue, task_id): if self.template_dict: return deepcopy(self.template_dict)