From af9f6fdbe4df17f78010a800ad3529ccf6ab1a30 Mon Sep 17 00:00:00 2001 From: Tim Repke <repke@mcc-berlin.net> Date: Sat, 10 Feb 2024 20:57:00 +0100 Subject: [PATCH] fix irr stuff --- server/api/routes/evaluation.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/server/api/routes/evaluation.py b/server/api/routes/evaluation.py index 5c04c43..b4157d3 100644 --- a/server/api/routes/evaluation.py +++ b/server/api/routes/evaluation.py @@ -7,6 +7,7 @@ from nacsos_data.db.schemas import AnnotationTracker, AssignmentScope, Annotatio AnnotationQuality from nacsos_data.models.annotation_quality import AnnotationQualityModel from nacsos_data.models.annotation_tracker import AnnotationTrackerModel, DehydratedAnnotationTracker +from nacsos_data.models.bot_annotations import BotAnnotationMetaDataBaseModel from nacsos_data.util.annotations.evaluation import get_new_label_batches from nacsos_data.util.annotations.evaluation.buscar import ( calculate_h0s_for_batches, @@ -62,6 +63,17 @@ async def get_project_scopes(permissions: UserPermissions = Depends(UserPermissi return assignment_scopes + resolution_scopes +@router.get('/resolutions', response_model=list[BotAnnotationMetaDataBaseModel]) +async def get_resolutions_for_scope(assignment_scope_id: str, + permissions: UserPermissions = Depends(UserPermissionChecker('annotations_read'))) \ + -> list[BotAnnotationMetaDataBaseModel]: + async with db_engine.session() as session: # type: AsyncSession + stmt = (select(BotAnnotationMetaData) + .where(BotAnnotationMetaData.assignment_scope_id == assignment_scope_id)) + rslt = (await session.execute(stmt)).scalars().all() + return [BotAnnotationMetaDataBaseModel.model_validate(r.__dict__) for r in rslt] + + async def read_tracker(session: AsyncSession, tracker_id: str | uuid.UUID, project_id: str | uuid.UUID | None = None) -> AnnotationTracker: stmt = (select(AnnotationTracker) -- GitLab