Support projects.get_hyperparam_values

This commit is contained in:
allegroai
2021-05-03 17:34:40 +03:00
parent 4cd4b2914d
commit 7f4ad0d1ca
6 changed files with 270 additions and 16 deletions

View File

@@ -14,6 +14,7 @@ from apiserver.apimodels.projects import (
ProjectReq,
ProjectTagsRequest,
ProjectTaskParentsRequest,
ProjectHyperparamValuesRequest,
)
from apiserver.bll.organization import OrgBLL, Tags
from apiserver.bll.project import ProjectBLL
@@ -397,6 +398,27 @@ def get_hyper_parameters(call: APICall, company_id: str, request: GetHyperParamR
}
@endpoint(
"projects.get_hyperparam_values",
min_version="2.13",
request_data_model=ProjectHyperparamValuesRequest,
)
def get_hyperparam_values(
call: APICall, company_id: str, request: ProjectHyperparamValuesRequest
):
total, values = task_bll.get_hyperparam_distinct_values(
company_id,
project_ids=request.projects,
section=request.section,
name=request.name,
allow_public=request.allow_public,
)
call.result.data = {
"total": total,
"values": values,
}
@endpoint(
"projects.get_task_tags", min_version="2.8", request_data_model=ProjectTagsRequest
)