updated

CogStack · May 21, 2019 · a60207d · a60207d
1 parent ccc1d96
commit a60207d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ann_converter.py b/ann_converter.py
@@ -37,7 +37,12 @@ def get_semehr_ann_label(ann):
             pref = ann.pref
         elif hasattr(ann, 'minor_type'):
             pref = ann.minor_type
-        return '%s%s(%s)' % (str_context, pref, ann.cui)
+        cui = ''
+        if hasattr(ann, 'cui'):
+            cui = ann.cui
+        elif hasattr(ann, 'major_type'):
+            cui = ann.major_type
+        return '%s%s(%s)' % (str_context, pref, cui)
 
     @staticmethod
     def to_eHOST(ann_doc, full_text=None, file_pattern='%s.txt', id_pattern='smehr-%s-%s',