ArneBinder / pytorch-ie

PyTorch-IE: State-of-the-art Information Extraction in PyTorch
MIT License
74 stars 8 forks source link

Fix `document_type` property of taskmodules #361

Closed ArneBinder closed 11 months ago

ArneBinder commented 11 months ago

With this PR, we check if the annotation layer names required in the taskmodules match the ones in the document types that they usually would request and request no document type in the case of a mismatch (but log a warning).

This also changes the default value for annotation in TransformerTextClassificationTaskModule to "label" to match the requested document type (this was missed in #359).