Revert "Reduce eud literals"

allenai · Jun 11, 2023 · 82ea5ac · 82ea5ac
1 parent 1dd64cc
commit 82ea5ac
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 18 deletions.
diff --git a/pybart/constants/constants.py b/pybart/constants/constants.py
diff --git a/pybart/constants/eud_literal_allowed_list_en.json b/pybart/constants/eud_literal_allowed_list_en.json
diff --git a/pybart/constants/eud_literal_allowed_list_he.json b/pybart/constants/eud_literal_allowed_list_he.json
diff --git a/pybart/graph_token.py b/pybart/graph_token.py
@@ -1,5 +1,4 @@
 from dataclasses import dataclass, field
-from .constants.constants import EUD_LITERAL_ALLOWED_LIST
 
 
 @dataclass
@@ -31,9 +30,7 @@ def to_str(self, remove_enhanced_extra_info, remove_bart_extra_info):
         eud = ""
         if self.eud is not None:
             if not remove_enhanced_extra_info:
-                if self.eud.lower() not in EUD_LITERAL_ALLOWED_LIST:
-                    self.eud = "_other"
-                eud = ":" + self.eud.lower()
+                eud = ":" + self.eud
 
         bart = ""
         if self.src is not None:

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="pybart-nlp",
-    version="3.4.5",
+    version="3.4.4",
     author="Aryeh Tiktinsky",
     author_email="aryehgigi@gmail.com",
     description="python converter from UD-tree to BART-graph representations",