You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

13 lines
428 B

from model_mommy import mommy
from projects.models import ProjectType
def make_annotation(task, doc, user, **kwargs):
annotation_model = {
ProjectType.DOCUMENT_CLASSIFICATION: "Category",
ProjectType.SEQUENCE_LABELING: "Span",
ProjectType.SEQ2SEQ: "TextLabel",
ProjectType.SPEECH2TEXT: "TextLabel",
}.get(task)
return mommy.make(annotation_model, example=doc, user=user, **kwargs)