Skip to content

Commit

Permalink
added python shebang, and correct typo with pd.read_csv
Browse files Browse the repository at this point in the history
  • Loading branch information
evanbiederstedt committed Aug 2, 2019
1 parent e05aef4 commit ca65bb1
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion neoantigen.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
#!/usr/bin/env python

from __future__ import print_function
from six.moves.configparser import ConfigParser

Expand Down Expand Up @@ -337,7 +339,7 @@ def main():
logger.info('Starting NetMHC 4.0...')
#####
# For netMHC-4 prediction, only predict on alleles for which data exists
netmhc_alleles = list(pd.read_csv(netmhc4_alleleslist, header=None, usecols=[0])[0], sep='\t')
netmhc_alleles = list(pd.read_csv(netmhc4_alleleslist, header=None, usecols=[0], sep='\t')[0])
alleles_for_prediction = list(set(netmhc_alleles) & set([x.replace(':', '') for x in hla_alleles]))
logger.info('Only predicting on the following HLA-alleles: ' + ','.join(sorted(set(alleles_for_prediction))))

Expand Down

0 comments on commit ca65bb1

Please sign in to comment.