Merge pull request #75 from ArneBinder/add_token_documents_with_multi…

…_spans add token document types with multi spans
ArneBinder · Mar 4, 2024 · e3c8fb2 · e3c8fb2
2 parents dea7466 + af5d2b4
commit e3c8fb2
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/src/pie_modules/documents.py b/src/pie_modules/documents.py
@@ -147,3 +147,30 @@ class TextDocumentWithLabeledMultiSpansBinaryRelationsAndLabeledPartitions(
     TextDocumentWithLabeledMultiSpansAndBinaryRelations,
 ):
     pass
+
+
+@dataclasses.dataclass
+class TokenDocumentWithLabeledMultiSpans(TokenBasedDocument):
+    labeled_multi_spans: AnnotationLayer[LabeledMultiSpan] = annotation_field(target="tokens")
+
+
+@dataclasses.dataclass
+class TokenDocumentWithLabeledMultiSpansAndLabeledPartitions(
+    TokenDocumentWithLabeledMultiSpans, TokenDocumentWithLabeledPartitions
+):
+    pass
+
+
+@dataclasses.dataclass
+class TokenDocumentWithLabeledMultiSpansAndBinaryRelations(TokenDocumentWithLabeledMultiSpans):
+    binary_relations: AnnotationLayer[BinaryRelation] = annotation_field(
+        target="labeled_multi_spans"
+    )
+
+
+@dataclasses.dataclass
+class TokenDocumentWithLabeledMultiSpansBinaryRelationsAndLabeledPartitions(
+    TokenDocumentWithLabeledMultiSpansAndLabeledPartitions,
+    TokenDocumentWithLabeledMultiSpansAndBinaryRelations,
+):
+    pass