Skip to content

Commit 0141550

Browse files
authored
Merge pull request #29 from bmichanderson/master
Fix paralogs_to_ref.py
2 parents fb5c3a1 + aa81423 commit 0141550

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

secapr/paralogs_to_ref.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,7 @@ def main(args):
9494
ref_index_info = os.path.join(root_dir,'reference_fasta_header_info.txt')
9595

9696
subdirs = list(os.walk(root_dir))[0][1]
97+
subdirs = [subdir for subdir in subdirs if not subdir.startswith('.')]
9798
for sample in subdirs:
9899
print('\nProcessing sample %s'%sample)
99100
# get the paths for the sample
@@ -104,7 +105,7 @@ def main(args):
104105
# read the data
105106
ref_index_df = pd.read_csv(ref_index_info,sep='\t',header=None)
106107
keys = ref_index_df[0].values
107-
values = ref_index_df[1].values
108+
values = [val.split()[0] for val in ref_index_df[1].values]
108109
id_ref_dict = dict(list(zip(keys,values)))
109110
ref_seqs = list(SeqIO.parse(ref_file, "fasta"))
110111
contig_seqs = list(SeqIO.parse(contig_file, "fasta"))

0 commit comments

Comments
 (0)