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

If possible, do not use api tasks object directly (use Task instead)

Browse Source
This commit is contained in:
allegroai 2020-09-15 18:13:07 +03:00
parent d36f6c26da
commit 2a249c41f0
1 changed files with 10 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
trains/backend_interface/task/development/stop_signal.py
View File

@ -1,5 +1,4 @@
from ....config import config from ....config import config
from ....backend_api.services import tasks
class TaskStopReason(object): class TaskStopReason(object):
@ -13,19 +12,18 @@ class TaskStopSignal(object):
_number_of_consecutive_reset_tests = 4 _number_of_consecutive_reset_tests = 4
# _unexpected_statuses = (
# tasks.TaskStatusEnum.closed,
# tasks.TaskStatusEnum.stopped,
# tasks.TaskStatusEnum.failed,
# tasks.TaskStatusEnum.published,
# tasks.TaskStatusEnum.completed,
# )
def __init__(self, task): def __init__(self, task):
from ....backend_interface import Task from ....backend_interface import Task
assert isinstance(task, Task) assert isinstance(task, Task)
self.task = task self.task = task
self._task_reset_state_counter = 0 self._task_reset_state_counter = 0
self._status_in_progress = str(Task.TaskStatusEnum.in_progress)
self._status_created = str(Task.TaskStatusEnum.created)
self._status_expected_statuses = (
str(Task.TaskStatusEnum.created),
str(Task.TaskStatusEnum.queued),
str(Task.TaskStatusEnum.in_progress),
)
def test(self): def test(self):
# noinspection PyBroadException # noinspection PyBroadException
@ -36,23 +34,17 @@ class TaskStopSignal(object):
status = str(status) status = str(status)
message = str(message) message = str(message)
if status == str(tasks.TaskStatusEnum.in_progress) and "stopping" in message: if status == self._status_in_progress and "stopping" in message:
# make sure we syn the entire task object # make sure we syn the entire task object
self.task.reload() self.task.reload()
return TaskStopReason.stopped return TaskStopReason.stopped
_expected_statuses = ( if status not in self._status_expected_statuses and "worker" not in message:
str(tasks.TaskStatusEnum.created),
str(tasks.TaskStatusEnum.queued),
str(tasks.TaskStatusEnum.in_progress),
)
if status not in _expected_statuses and "worker" not in message:
# make sure we syn the entire task object # make sure we syn the entire task object
self.task.reload() self.task.reload()
return TaskStopReason.status_changed return TaskStopReason.status_changed
if status == str(tasks.TaskStatusEnum.created): if status == self._status_created:
self._task_reset_state_counter += 1 self._task_reset_state_counter += 1
if self._task_reset_state_counter >= self._number_of_consecutive_reset_tests: if self._task_reset_state_counter >= self._number_of_consecutive_reset_tests: