fix attribute to id mapping for format change due to #28

GeneMANIA · Jul 21, 2015 · ad2049e · ad2049e
1 parent d1b948e
commit ad2049e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/builder/map_attributes_to_ids.py b/builder/map_attributes_to_ids.py
@@ -9,7 +9,7 @@
 
 def main(filename, desc, mapping, output):
 
-    desc = pd.read_csv(desc, sep='\t', header=False, na_filter=False, names = ['ATTRIBUTE_ID', 'ATTRIBUTE_SYMBOL', 'DESCRIPTION'])
+    desc = pd.read_csv(desc, sep='\t', header=False, na_filter=False, names = ['ATTRIBUTE_ID', 'ATTRIBUTE_SYMBOL', 'ATTRIBUTE_NAME', 'DESCRIPTION'])
     desc['ATTRIBUTE_SYMBOL'] = desc['ATTRIBUTE_SYMBOL'].str.upper()
 
     symbols = pd.read_csv(mapping, sep='\t', header=None, na_filter=False, names=['NODE_ID', 'GENE_SYMBOL', 'SOURCE'])
@@ -42,4 +42,4 @@ def main(filename, desc, mapping, output):
     parser.add_argument('output', help='output file')
 
     args = parser.parse_args()
-    main(args.filename, args.desc, args.mapping, args.output)
+    main(args.filename, args.desc, args.mapping, args.output)