fix whitespace in unii display_name

biothings · May 29, 2024 · 858e4fa · 858e4fa
1 parent 19ed808
commit 858e4fa
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/hub/dataload/sources/unii/unii_parser.py b/src/hub/dataload/sources/unii/unii_parser.py
@@ -63,9 +63,14 @@ def load_data(input_file):
                 record['unii']['ncit_description'] = ncit_descriptions[ncit_id]
         if isinstance(record['unii'], dict):
             del record['unii']['_id']
+            if 'display name' in record['unii']:
+                record['unii']['display_name'] = record['unii'].pop(
+                    'display name')
         else:
             for subr in record['unii']:
                 del subr['_id']
+                if 'display name' in subr:
+                    subr['display_name'] = subr.pop('display name')
 
         # convert fields to integer
         record = int_convert(record, include_keys=['unii.pubchem'])