Merge pull request #326 from UnB-KnEDLe/highlight_patch

Highlight structure now returns entity type
UnB-KnEDLe · Mar 18, 2023 · 702b4bf · 702b4bf
2 parents 228494d + 65f2d62
commit 702b4bf
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/dodfminer/__version__.py b/dodfminer/__version__.py
@@ -1,4 +1,4 @@
-version_info = (1, 4, 4)
+version_info = (1, 4, 5)
 # format:
 # ('dodf_major', 'dodf_minor', 'dodf_patch')
 

diff --git a/dodfminer/extract/polished/acts/base.py b/dodfminer/extract/polished/acts/base.py
@@ -223,7 +223,8 @@ def highlight_dataframe(self):
                 current_ent = {
                     "name": [],
                     "start": None,
-                    "end": None
+                    "end": None,
+                    "type": None
                 }
 
                 if "B-" in IOB[i]:
@@ -233,6 +234,7 @@ def highlight_dataframe(self):
 
                     current_ent["start"] = len(aux_text_string) + 1
                     current_ent["name"].append(text_split[i])
+                    current_ent["type"] = entity_name
 
                     i += 1
 

diff --git a/dodfminer/extract/polished/acts/base_contratos.py b/dodfminer/extract/polished/acts/base_contratos.py
@@ -149,7 +149,8 @@ def highlight_dataframe(self):
           current_ent = {
               "name": [],
               "start": None,
-              "end": None
+              "end": None,
+              "type": None
           }
 
           if "B-" in IOB[i]:
@@ -159,6 +160,7 @@ def highlight_dataframe(self):
 
               current_ent["start"] = len(aux_text_string) + 1
               current_ent["name"].append(text_split[i])
+              current_ent["type"] = entity_name
 
               i += 1