ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-04 02:57:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add original_task property to models

Browse Source 
Change `task` property to return connected task
This commit is contained in:
allegroai 2024-07-21 14:14:15 +03:00
parent a59b40217e
commit fa0ba104c4
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
clearml/model.py
View File

@ -324,11 +324,22 @@ class BaseModel(object):
def task(self): def task(self):
# type: () -> str # type: () -> str
""" """
Return the creating task ID Return the task ID connected to this model. If not task is connected,
return the ID of the task that originally created this model.
:return: The Task ID (str) :return: The Task ID (str)
""" """
return self._task.id if self._task else self._get_base_model().task return self._task.id if self._task else self.original_task
@property
def original_task(self):
# type: () -> str
"""
Return the ID of the task that created this model.
:return: The Task ID (str)
"""
return self._get_base_model().task
@property @property
def url(self): def url(self):