Skip to content
Snippets Groups Projects
Commit ce485ed9 authored by Studer Gabriel's avatar Studer Gabriel
Browse files

Set ss_pred and ss_conf to None if not available in a3m/hhm parsing

parent 318a1bde
No related branches found
No related tags found
No related merge requests found
...@@ -302,6 +302,7 @@ def ParseA3M(a3m_file): ...@@ -302,6 +302,7 @@ def ParseA3M(a3m_file):
:return: Dictionary containing "ss_pred" (:class:`list`), "ss_conf" :return: Dictionary containing "ss_pred" (:class:`list`), "ss_conf"
(:class:`list`) and "msa" (:class:`~ost.seq.AlignmentHandle`). (:class:`list`) and "msa" (:class:`~ost.seq.AlignmentHandle`).
If not available, "ss_pred" and "ss_conf" entries are set to None.
''' '''
profile_dict = dict() profile_dict = dict()
state = 'NONE' state = 'NONE'
...@@ -331,12 +332,16 @@ def ParseA3M(a3m_file): ...@@ -331,12 +332,16 @@ def ParseA3M(a3m_file):
elif state == 'msa': elif state == 'msa':
msa_seq[len(msa_seq)-1] += line.rstrip() msa_seq[len(msa_seq)-1] += line.rstrip()
profile_dict['ss_pred'] = list() if len(pred_seq_txt) > 0:
profile_dict['ss_conf'] = list() profile_dict['ss_pred'] = list()
for i in range(0, len(pred_seq_txt)): profile_dict['ss_conf'] = list()
profile_dict['ss_pred'].append(pred_seq_txt[i]) for i in range(0, len(pred_seq_txt)):
profile_dict['ss_conf'].append(int(conf_seq_txt[i])) profile_dict['ss_pred'].append(pred_seq_txt[i])
profile_dict['ss_conf'].append(int(conf_seq_txt[i]))
else:
profile_dict['ss_pred'] = None
profile_dict['ss_conf'] = None
# post processing # post processing
# MSA # MSA
profile_dict['msa'] = None profile_dict['msa'] = None
...@@ -374,6 +379,7 @@ def ParseHHM(profile): ...@@ -374,6 +379,7 @@ def ParseHHM(profile):
:return: Dictionary containing "ss_pred" (:class:`list`), "ss_conf" :return: Dictionary containing "ss_pred" (:class:`list`), "ss_conf"
(:class:`list`), "msa" (:class:`~ost.seq.AlignmentHandle`) and (:class:`list`), "msa" (:class:`~ost.seq.AlignmentHandle`) and
"consensus" (:class:`~ost.seq.SequenceHandle`). "consensus" (:class:`~ost.seq.SequenceHandle`).
If not available, "ss_pred" and "ss_conf" entries are set to None.
''' '''
profile_dict = dict() profile_dict = dict()
state = 'NONE' state = 'NONE'
...@@ -416,11 +422,15 @@ def ParseHHM(profile): ...@@ -416,11 +422,15 @@ def ParseHHM(profile):
elif state == 'consensus': elif state == 'consensus':
consensus_txt += line.rstrip() consensus_txt += line.rstrip()
profile_dict['ss_pred'] = list() if len(pred_seq_txt) > 0:
profile_dict['ss_conf'] = list() profile_dict['ss_pred'] = list()
for i in range(0, len(pred_seq_txt)): profile_dict['ss_conf'] = list()
profile_dict['ss_pred'].append(pred_seq_txt[i]) for i in range(0, len(pred_seq_txt)):
profile_dict['ss_conf'].append(int(conf_seq_txt[i])) profile_dict['ss_pred'].append(pred_seq_txt[i])
profile_dict['ss_conf'].append(int(conf_seq_txt[i]))
else:
profile_dict['ss_pred'] = None
profile_dict['ss_conf'] = None
# post processing # post processing
# MSA # MSA
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment