minor fixes

ArneBinder · Dec 19, 2023 · 172fc14 · 172fc14
1 parent 612466b
commit 172fc14
Show file tree

Hide file tree

Showing 3 changed files with 175 additions and 184 deletions.
diff --git a/dataset_builders/pie/abstrct/README.md b/dataset_builders/pie/abstrct/README.md
@@ -46,9 +46,9 @@ assert isinstance(doc, builders.brat.BratDocumentWithMergedSpans)
 The dataset provides document converters for the following target document types:
 
 - `pytorch_ie.documents.TextDocumentWithLabeledSpansAndBinaryRelations`
-  - `LabeledSpans`, converted from `BratDocument`'s `spans`
+  - `LabeledSpans`, converted from `BratDocumentWithMergedSpans`'s `spans`
     - labels: `MajorClaim`, `Claim`, `Premise`
-  - `BinraryRelations`, converted from `BratDocument`'s `relations`
+  - `BinraryRelations`, converted from `BratDocumentWithMergedSpans`'s `relations`
     - labels:  `Support`, `Partial-Attack`, `Attack`
 
 See [here](https://github.com/ChristophAlt/pytorch-ie/blob/main/src/pytorch_ie/documents.py) for the document type

diff --git a/dataset_builders/pie/abstrct/abstrct.py b/dataset_builders/pie/abstrct/abstrct.py
@@ -27,7 +27,7 @@ class AbstRCT(BratBuilder):
     # we need to add None to the list of dataset variants to support the default dataset variant
     BASE_BUILDER_KWARGS_DICT = {
         dataset_variant: {"url": URL, "split_paths": SPLIT_PATHS}
-        for dataset_variant in ["default", "merge_fragmented_spans", None]
+        for dataset_variant in ["default", None]
     }
 
     DOCUMENT_CONVERTERS = {