Fix document types with partitions (#349)

* fix TextDocumentWithLabeledEntitiesRelationsAndLabeledPartitions * fix TextDocumentWithEntitiesRelationsAndLabeledPartitions
ArneBinder · Sep 23, 2023 · f697962 · f697962
1 parent 119a725
commit f697962
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/src/pytorch_ie/documents.py b/src/pytorch_ie/documents.py
@@ -98,7 +98,9 @@ class TextDocumentWithLabeledEntitiesAndRelations(TextDocumentWithLabeledEntitie
 
 @dataclasses.dataclass
 class TextDocumentWithLabeledEntitiesRelationsAndLabeledPartitions(
-    TextDocumentWithLabeledEntitiesAndRelations, TextDocumentWithLabeledPartitions
+    TextDocumentWithLabeledEntitiesAndLabeledPartitions,
+    TextDocumentWithLabeledEntitiesAndRelations,
+    TextDocumentWithLabeledPartitions,
 ):
     pass
 
@@ -108,8 +110,17 @@ class TextDocumentWithEntitiesAndRelations(TextDocumentWithEntities):
     relations: AnnotationList[BinaryRelation] = annotation_field(target="entities")
 
 
+@dataclasses.dataclass
+class TextDocumentWithEntitiesAndLabeledPartitions(
+    TextDocumentWithEntities, TextDocumentWithLabeledPartitions
+):
+    pass
+
+
 @dataclasses.dataclass
 class TextDocumentWithEntitiesRelationsAndLabeledPartitions(
-    TextDocumentWithEntitiesAndRelations, TextDocumentWithLabeledPartitions
+    TextDocumentWithEntitiesAndLabeledPartitions,
+    TextDocumentWithEntitiesAndRelations,
+    TextDocumentWithLabeledPartitions,
 ):
     pass