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

Optimize queues.get_next_task to retrieve required task fields only

Browse Source
This commit is contained in:
allegroai 2023-05-25 19:21:24 +03:00
parent dff2ed34e8
commit 3729afe014
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apiserver/services/queues.py
View File

@ -171,7 +171,7 @@ def get_next_task(call: APICall, company_id, request: GetNextTaskRequest):
if entry: if entry:
data = {"entry": entry.to_proper_dict()} data = {"entry": entry.to_proper_dict()}
if request.get_task_info: if request.get_task_info:
task = Task.objects(id=entry.task).first() task = Task.objects(id=entry.task).only("company", "user").first()
if task: if task:
data["task_info"] = {"company": task.company, "user": task.user} data["task_info"] = {"company": task.company, "user": task.user}