#328 fixed

nicolay-r · Jun 7, 2022 · 8c38e5a · 8c38e5a
1 parent 040d6b3
commit 8c38e5a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/arekit/contrib/source/brat/entities/reader.py b/arekit/contrib/source/brat/entities/reader.py
@@ -14,6 +14,12 @@ def extract_entities(input_file):
             line = line.decode('utf-8')
 
             args = line.split()
+
+            record_type = args[0][1]
+            # Entities (objects) are prefixed with `T`
+            if record_type != 'T':
+                continue
+
             e_id = int(args[0][1:])
             e_str_type = args[1]
             e_begin = int(args[2])