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

Pass task_id when resolving k8s template

Browse Source
This commit is contained in:
allegroai 2024-04-01 11:37:01 +03:00
parent 3d43240c8f
commit 43443ccf08
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml_agent/glue/k8s.py
View File

@ -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)