Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -169,10 +169,8 @@ 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')]
-
-
         for dp in data.index:
-            if (data.iloc[dp, 'wt_sequence_match'] == 'nan') & (data.iloc[dp, 'whichIsoform'] == 'nan'):
+            if (data.at[dp, 'wt_sequence_match'] == 'nan') & (data.at[dp, 'whichIsoform'] == 'nan'):
                 st.write(f'Wrong input aminoacid for datapoint: {data.iloc[dp, 'datapoint']}')
         data = None