Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -256,8 +256,8 @@ def pdb(input_set, mode, impute):
                     modbase = pd.DataFrame(columns = SIMPLE_COLS)
     
             else:
-                no_modbase_models_updated = pd.DataFrame()
-                modbase= pd.DataFrame()
+                no_modbase_models_updated = pd.DataFrame(columns = SIMPLE_COLS)
+                modbase= pd.DataFrame(columns = SIMPLE_COLS)
     
             COLS = ['uniprotID', 'wt', 'pos', 'mut', 'datapoint', 'composition', 'polarity', 'volume', 'granthamScore', 'domain', 'domStart', 'domEnd', 'distance',
                     'region', 'crosslink', 'peptide', 'disulfide', 'signalPeptide', 'propeptide', 'naturalVariant', 'nucleotideBinding', 'modifiedResidue', 'site',
@@ -272,9 +272,13 @@ def pdb(input_set, mode, impute):
                 no_modbase_models_updated = pd.DataFrame()
             st.write('no_modbase_models_updated')
             st.write(no_modbase_models_updated)
+            st.write(len(no_modbase_models_updated))
+            
 
             st.write('modbase')
             st.write(modbase)
+            st.write(len(modbase))
+
             no_modbase_models_updated = no_modbase_models_updated[~no_modbase_models_updated.datapoint.isin(modbase.datapoint.to_list())]
             no_modbase_models_updated = no_modbase_models_updated[['uniprotID', 'wt', 'pos', 'mut', 'datapoint']]
             no_modbase_models_updated.pos = no_modbase_models_updated.pos.astype(int)