update gene list reading to be compatible w python3

SystemsGenetics · May 1, 2019 · c8e9988 · c8e9988
1 parent 597186c
commit c8e9988
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/classify.py b/scripts/classify.py
@@ -224,7 +224,7 @@ def full_classification(data, total_gene_list, config, out_file, kfold_val=10, v
 	# load the data
 	print('loading genetic data...')
 	gtex_gct_flt = np.load(args.dataset)
-	total_gene_list = np.load(args.gene_list)
+	total_gene_list = np.load(args.gene_list).astype(np.str)
 	data = load_data(args.sample_json, gtex_gct_flt)
 
 	# load interaction data, if passed

diff --git a/scripts/gene-oracle.py b/scripts/gene-oracle.py
@@ -181,7 +181,7 @@ def generate_new_subsets_wo_clustering(file, data, total_gene_list, genes, max_e
 
 	print('loading genetic data...')
 	gtex_gct_flt = np.load(args.dataset)
-	total_gene_list = np.load(args.gene_list, encoding='ASCII')
+	total_gene_list = np.load(args.gene_list).astype(np.str)
 	print('done')
 
 	data = load_data(args.sample_json, gtex_gct_flt)