#328 and #329 related fix.

nicolay-r · Jun 14, 2022 · b1c797c · b1c797c
1 parent 471a749
commit b1c797c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/arekit/contrib/source/brat/annot.py b/arekit/contrib/source/brat/annot.py
@@ -28,7 +28,7 @@ def handle_entity(args):
     @staticmethod
     def handle_relation(args):
 
-        e_id = int(args[0][1:])
+        e_id = args[0][1:]
 
         rel_type = args[1]
         source_id = args[2].split(':')[1]
@@ -51,7 +51,7 @@ def parse_annotations(input_file, encoding='utf-8'):
 
             args = line.split()
 
-            record_type = args[0][1]
+            record_type = args[0][0]
 
             # Entities (objects) are prefixed with `T`
             if record_type == "T":