added default predictions in predefined examples

predefined_example.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 
 from span_dataclass_converters import get_ner_spans_from_annotations
 
@@ -7,6 +7,7 @@ from span_dataclass_converters import get_ner_spans_from_annotations
 class PredefinedExample:
     text: str
     gt_labels: dict
+    default_predictions: list = field(default_factory=list)
     # gt_spans: list
     # predictions: list
 
@@ -19,7 +20,7 @@ class PredefinedExample:
 
     @property
     def predictions(self):
-        return [self.gt_spans]
+        return [self.gt_spans] + self.default_predictions
 
     @property
     def tags(self):
@@ -36,6 +37,29 @@ small_example = PredefinedExample(
             {"start": 63, "end": 72, "label": "mucolytic"},
         ],
     },
+    default_predictions=[
+        [
+            {
+                "start": 26,
+                "end": 41,
+                "label": "Disease",
+                "span_text": "with bronchitis",
+            },
+            {"start": 61, "end": 72, "label": "Drug", "span_text": "a mucolytic"},
+        ],
+        [
+            {"start": 31, "end": 41, "label": "Drug", "span_text": "bronchitis"},
+            {"start": 63, "end": 72, "label": "Drug", "span_text": "mucolytic"},
+        ],
+        [
+            {
+                "start": 31,
+                "end": 72,
+                "label": "Disease",
+                "span_text": "bronchitis and was prescribed a mucolytic",
+            }
+        ],
+    ],
 )
 
 big_example = PredefinedExample(
@@ -57,6 +81,33 @@ big_example = PredefinedExample(
             {"start": 46, "end": 63, "label": "flu like symptoms"},
         ],
     },
+    default_predictions=[
+        [
+            {"start": 29, "end": 41, "label": "Symptoms", "span_text": "stomach pain"},
+            {"start": 46, "end": 49, "label": "Symptoms", "span_text": "flu"},
+            {
+                "start": 120,
+                "end": 136,
+                "label": "Disease",
+                "span_text": "acute bronchitis",
+            },
+            {"start": 213, "end": 222, "label": "Drug", "span_text": "mucolytic"},
+            {"start": 234, "end": 245, "label": "Drug", "span_text": "paracetamol"},
+        ],
+        [
+            {"start": 29, "end": 41, "label": "Symptoms", "span_text": "stomach pain"},
+            {"start": 46, "end": 49, "label": "Disease", "span_text": "flu"},
+            {
+                "start": 120,
+                "end": 136,
+                "label": "Disease",
+                "span_text": "acute bronchitis",
+            },
+            {"start": 213, "end": 222, "label": "Drug", "span_text": "mucolytic"},
+            {"start": 234, "end": 245, "label": "Drug", "span_text": "paracetamol"},
+            {"start": 250, "end": 260, "label": "Symptoms", "span_text": "body pains"},
+        ],
+    ],
 )
 
 EXAMPLES = [small_example, big_example]