ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-06-26 23:15:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add force parameter to tasks.enqueue

Browse Source
This commit is contained in:
allegroai 2021-07-25 14:27:46 +03:00
parent cb253cff9e
commit 677bb3ba6d
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apiserver/bll/task/task_operations.py
View File

@ -109,6 +109,7 @@ def enqueue_task(
status_message: str, status_message: str,
status_reason: str, status_reason: str,
validate: bool = False, validate: bool = False,
force: bool = False,
) -> Tuple[int, dict]: ) -> Tuple[int, dict]:
if not queue_id: if not queue_id:
# try to get default queue # try to get default queue
@ -128,6 +129,7 @@ def enqueue_task(
status_reason=status_reason, status_reason=status_reason,
status_message=status_message, status_message=status_message,
allow_same_state_transition=False, allow_same_state_transition=False,
force=force,
).execute(enqueue_status=task.status) ).execute(enqueue_status=task.status)
try: try:

1
apiserver/services/tasks.py
View File

@ -854,6 +854,7 @@ def enqueue(call: APICall, company_id, request: EnqueueRequest):
queue_id=request.queue, queue_id=request.queue,
status_message=request.status_message, status_message=request.status_message,
status_reason=request.status_reason, status_reason=request.status_reason,
force=request.force,
) )
call.result.data_model = EnqueueResponse(queued=queued, **res) call.result.data_model = EnqueueResponse(queued=queued, **res)