ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-04-18 13:15:37 +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)) print("Kubernetes scheduling task id={}".format(task_id))
try: 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: except Exception as ex:
print("ERROR: Failed resolving template (skipping): {}".format(ex)) print("ERROR: Failed resolving template (skipping): {}".format(ex))
return return
@ -1106,7 +1106,7 @@ class K8sIntegration(Worker):
self._session, queue=queue, get_task_info=get_task_info 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: if self.template_dict:
return deepcopy(self.template_dict) return deepcopy(self.template_dict)