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

Add Task.set_archived() / Task.get_archived()

Browse Source
This commit is contained in:
allegroai 2021-01-10 13:01:04 +02:00
parent c9026ad4a1
commit d39f7bc195
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
clearml/backend_interface/task/task.py
View File

@ -1367,6 +1367,27 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
self._set_task_property("task_type", str(task_type)) self._set_task_property("task_type", str(task_type))
self._edit(type=task_type) self._edit(type=task_type)
def set_archived(self, archive):
# type: (bool) -> ()
"""
Archive the Task or remove it from the archived folder.
:param archive: If True archive the Task, If False make sure it is removed from the archived folder
"""
with self._edit_lock:
system_tags = list(set(self.get_system_tags()) | {self.archived_tag}) \
if archive else list(set(self.get_system_tags()) - {self.archived_tag})
self.set_system_tags(system_tags)
def get_archived(self):
# type: () -> bool
"""
Return the Archive state of the Task
:return: If True the Task is archived, otherwise it is not.
"""
return self.archived_tag in self.get_system_tags()
def set_initial_iteration(self, offset=0): def set_initial_iteration(self, offset=0):
# type: (int) -> int # type: (int) -> int
""" """
@ -1734,7 +1755,7 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
status = self._data.status if self._data and self._reload_skip_flag else self.data.status status = self._data.status if self._data and self._reload_skip_flag else self.data.status
if status not in (tasks.TaskStatusEnum.created, tasks.TaskStatusEnum.in_progress): if status not in (tasks.TaskStatusEnum.created, tasks.TaskStatusEnum.in_progress):
# the exception being name/comment that we can always change. # the exception being name/comment that we can always change.
if kwargs and all(k in ('name', 'comment', 'tags') for k in kwargs.keys()): if kwargs and all(k in ('name', 'comment', 'tags', 'system_tags') for k in kwargs.keys()):
pass pass
else: else:
raise ValueError('Task object can only be updated if created or in_progress') raise ValueError('Task object can only be updated if created or in_progress')