Spaces:
Running
Running
File size: 1,011 Bytes
e1df50c 7a295c7 e1df50c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
# scripts/assign_interval_to_annotator.py
from utils.database import get_db
from data.repository.annotator_repo import AnnotatorRepo
from data.repository.annotation_interval_repo import AnnotationIntervalRepo
from utils.logger import Logger
log = Logger()
START_IDX = 101
END_IDX = 120
ANNOTATOR_NAME = "vargha"
with get_db() as db:
annot_repo = AnnotatorRepo(db)
interval_repo = AnnotationIntervalRepo(db)
annotator = annot_repo.get_annotator_by_name(ANNOTATOR_NAME)
if not annotator:
log.info(f"Annotator '{annotator}' not found;")
try:
interval = interval_repo.assign_interval_to_annotator(
annotator_id=annotator.id,
start_idx=START_IDX,
end_idx=END_IDX,
)
log.info(
f"Interval [{interval.start_index}, {interval.end_index}] "
f"successfully assigned to '{annotator.name}' (id={annotator.id})"
)
except ValueError as err:
log.error(f"Could not assign interval: {err}")
|