ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 17:17:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add TrainsJob.is_completed()/is_aborted()

Browse Source
This commit is contained in:
allegroai 2020-09-01 17:45:08 +03:00
parent fd165f7126
commit d148fe8042
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
trains/automation/job.py
View File

@ -35,10 +35,12 @@ class TrainsJob(object):
self.task = Task.clone(base_task_id, parent=parent or base_task_id, **kwargs)
if tags:
self.task.set_tags(list(set(self.task.get_tags()) | set(tags)))
self.task_parameter_override = None
if parameter_override:
params = self.task.get_parameters_as_dict()
params.update(parameter_override)
self.task.set_parameters_as_dict(params)
self.task_parameter_override = dict(**parameter_override)
if task_overrides:
# todo: make sure it works
# noinspection PyProtectedMember
@ -217,6 +219,24 @@ class TrainsJob(object):
"""
return self.task.status in (Task.TaskStatusEnum.failed, )
def is_completed(self):
# type: () -> bool
"""
Return True, if job is has executed and completed successfully
:return: True the task is currently in completed or published state
"""
return self.task.status in (Task.TaskStatusEnum.completed, Task.TaskStatusEnum.published)
def is_aborted(self):
# type: () -> bool
"""
Return True, if job is has executed and aborted
:return: True the task is currently in aborted state
"""
return self.task.status in (Task.TaskStatusEnum.stopped, )
def is_pending(self):
# type: () -> bool
"""