from model.bert import BertForRelationAwareClassification

from model.model_output import RelationAwareClassificationModelOutput


MODEL_TYPE_TO_CLASS = {
    'bert': BertForRelationAwareClassification,
}