Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -1635,8 +1635,7 @@ def pdb(input_set, mode, impute):
              'glycosylation_dist', 'propeptide_dist']]
         ready = data.copy()
         # Imputation
-        if (impute == 'True') or (impute == 'true'):
-            st.write('HERE', impute)
+        if (impute == 'True') or (impute == 'true') or (impute == True):
             filler = [17.84, 30.8, 24.96, 13.12, 23.62, 18.97, 20.87, 29.59, 20.7, 12.7, 22.85, 17.21, 9.8, 9, 15.99, 16.82,
                       20.46, 24.58, 9.99, 17.43, 20.08, 30.91, 20.86, 22.14, 21.91, 28.45, 17.81, 25.12, 20.33, 22.36]
             col_index = 0
@@ -1650,8 +1649,6 @@ def pdb(input_set, mode, impute):
         elif (impute == 'False') or (impute == 'false'):
             st.write('DEĞİL', impute)
             pass
-        else:
-            st.write('aaa', impute)
         ready = ready.replace({'nan': np.NaN})
         ready.to_csv(path_to_output_files / 'featurevector_pdb.txt', sep='\t', index=False)
         if len(ready) == 0: