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

Support workers filtering with tags

Browse Source
This commit is contained in:
allegroai
2022-07-08 17:37:33 +03:00
parent b41ab8c550
commit b2feafac09
4 changed files with 473 additions and 450 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
apiserver/services/workers.py
View File

@@ -41,7 +41,9 @@ worker_bll = WorkerBLL()
)
def get_all(call: APICall, company_id: str, request: GetAllRequest):
call.result.data_model = GetAllResponse(
workers=worker_bll.get_all_with_projection(company_id, request.last_seen)
workers=worker_bll.get_all_with_projection(
company_id, request.last_seen, tags=request.tags
)
)
@@ -72,7 +74,9 @@ def unregister(call: APICall, company_id, req_model: WorkerRequest):
worker_bll.unregister_worker(company_id, call.identity.user, req_model.worker)
@endpoint("workers.status_report", min_version="2.4", request_data_model=StatusReportRequest)
@endpoint(
"workers.status_report", min_version="2.4", request_data_model=StatusReportRequest
)
def status_report(call: APICall, company_id, request: StatusReportRequest):
worker_bll.status_report(
company_id=company_id,