#507 fixed label formatter

nicolay-r · Jul 29, 2023 · 5f966f2 · 5f966f2
1 parent dc94c2c
commit 5f966f2
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 11 deletions.
diff --git a/arekit/contrib/source/nerel/labels.py b/arekit/contrib/source/nerel/labels.py
@@ -41,18 +41,10 @@ class PositiveTo(Label):
     pass
 
 
-class ALTERNATIVE_NAME(Label):
-    pass
-
-
 class STATE_BELONGS_TO(Label):
     pass
 
 
-class ORIGINS_FROM(Label):
-    pass
-
-
 class ABBREVIATION(Label):
     pass
 

diff --git a/arekit/contrib/utils/pipelines/sources/nerel/labels_fmt.py b/arekit/contrib/utils/pipelines/sources/nerel/labels_fmt.py
@@ -15,11 +15,46 @@ def __init__(self):
             "POS_STATE_FROM": labels.PosStateFrom,
             "NEGATIVE_TO": labels.NegativeTo,
             "POSITIVE_TO": labels.PositiveTo,
-            "STATE_BELONGS_TO": labels.StateBelongsTo,
+            "STATE_BELONGS_TO": labels.STATE_BELONGS_TO,
             "POS_AUTHOR_FROM": labels.PosAuthorFrom,
             "NEG_AUTHOR_FROM": labels.NegAuthorFrom,
-            "ALTERNATIVE_NAME": labels.AlternativeName,
-            "ORIGINS_FROM": labels.OriginsFrom
+            "ALTERNATIVE_NAME": labels.ALTERNATIVE_NAME,
+            "ORIGINS_FROM": labels.ORIGINS_FROM,
+            "START_TIME": labels.START_TIME,
+            "OWNER_OF": labels.OWNER_OF,
+            "SUBEVENT_OF": labels.SUBEVENT_OF,
+            "PARENT_OF": labels.PARENT_OF,
+            "SUBORDINATE_OF": labels.SUBORDINATE_OF,
+            "PART_OF": labels.PART_OF,
+            "TAKES_PLACE_IN": labels.TAKES_PLACE_IN,
+            "PARTICIPANT_IN": labels.PARTICIPANT_IN,
+            "WORKPLACE": labels.WORKPLACE,
+            "PENALIZED_AS": labels.PENALIZED_AS,
+            "WORKS_AS": labels.WORKS_AS,
+            "PLACE_OF_DEATH": labels.PLACE_OF_DEATH,
+            "PLACE_OF_BIRTH": labels.PLACE_OF_BIRTH,
+            "HAS_CAUSE": labels.HAS_CAUSE,
+            "AWARDED_WITH": labels.AWARDED_WITH,
+            "CAUSE_OF_DEATH": labels.CAUSE_OF_DEATH,
+            "CONVICTED_OF": labels.CONVICTED_OF,
+            "DATE_DEFUNCT_IN": labels.DATE_DEFUNCT_IN,
+            "DATE_FOUNDED_IN": labels.DATE_FOUNDED_IN,
+            "DATE_OF_BIRTH": labels.DATE_OF_BIRTH,
+            "DATE_OF_CREATION": labels.DATE_OF_CREATION,
+            "DATE_OF_DEATH": labels.DATE_OF_DEATH,
+            "END_TIME": labels.END_TIME,
+            "EXPENDITURE": labels.EXPENDITURE,
+            "FOUNDED_BY": labels.FOUNDED_BY,
+            "KNOWS": labels.KNOWS,
+            "RELATIVE": labels.RELATIVE,
+            "LOCATED_IN": labels.LOCATED_IN,
+            "RELIGION_OF": labels.RELIGION_OF,
+            "MEDICAL_CONDITION": labels.MEDICAL_CONDITION,
+            "SCHOOLS_ATTENDED": labels.SCHOOLS_ATTENDED,
+            "MEMBER_OF": labels.MEMBER_OF,
+            "SIBLING": labels.SIBLING,
+            "ORGANIZES": labels.ORGANIZES,
+            "SPOUSE": labels.SPOUSE
         }
 
         super(NerelAnyLabelFormatter, self).__init__(stol=stol)