Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -256,7 +256,13 @@ def pdb(input_set, mode, impute):
                     modbase = isZeroDistance(modbase)
                     modbase = match3DModels(modbase)
                     modbase = selectMaxAnnot(modbase)
+                    modbase['quality_score'] = modbase['quality_score'].astype(float)
+                    modbase['distance'] = modbase['distance'].astype(float)
+                    modbase['hitTotal'] = modbase['hitTotal'].astype(float)
+                    modbase['annotTotal'] = modbase['annotTotal'].astype(float)
                     modbase = modbase.sort_values(by=['datapoint', 'quality_score', 'distance','hitTotal', 'annotTotal'], ascending=[True, True, True, True, True])
+                    st.write('SORTED')
+                    st.write(modbase)
                     modbase = modbase.drop_duplicates(['datapoint'])
                     modbase.replace({'[]': np.NaN, 'hit': 0.0}, inplace=True)
                     st.write('modbase SIMP')