Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -171,10 +171,7 @@ def pdb(input_set, mode, impute):
         data_size = len(data.drop_duplicates(['datapoint']))
         not_match_in_uniprot = data[(data.uniprotSequence == 'nan') | (data.wt_sequence_match == 'nan')]
         uniprot_matched = data[(data.uniprotSequence != 'nan') & (data.wt_sequence_match != 'nan')]
-        st.write(uniprot_matched)
-        st.write(not_match_in_uniprot)
-        if (len(uniprot_matched) == 0) & (len(not_match_in_uniprot) == 0):
-            st.write('Please check the input aminoacid at the selected position.')
+
 
         
         data = None
@@ -493,9 +490,10 @@ def pdb(input_set, mode, impute):
         print('--PDB structures not found for %d datapoints.' % len(no_pdb.drop_duplicates(['datapoint'])))
         print('--%d will be searched in Swiss-Model database.\n' % (
                 len(yes_pdb_no_match.drop_duplicates(['datapoint'])) + len(no_pdb.drop_duplicates(['datapoint']))))
-        st.write('dfM')
-        st.write(dfM)
-        st.write(dfNM)
+
+        if (len(dfM) == 0) & (len(dfNM) == 0):
+            st.write('Please check the input aminoacid at the selected position.')
+
         dfM = None
         dfNM = None
         aligned_nm = None