Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -40,9 +40,9 @@ from add_alignment import *
 from manage_files import *
 from add_3Dalignment import *
 from add_sasa import *
-from standard import standardize
-from standard import finalTouch
+from standard import *
 from add_interface_pos import *
+from standard import *
 from utils import *
 from pdbMapping import *
 from uniprotSequenceMatch import uniprotSequenceMatch
@@ -309,7 +309,6 @@ def pdb(input_set, mode, impute):
     
                 print('Extracting interface residues...\n')
                 data_interface = pd.read_csv(path_to_interfaces, sep='\t')
-                
     
                 positions = get_interface_positions(data_interface, 'P1', 'P2')
     
@@ -318,15 +317,10 @@ def pdb(input_set, mode, impute):
                 for key, val in positions.items():
                     k = pd.Series((key, str(list(set(val)))))
                     interface_dataframe = interface_dataframe.append(k, ignore_index=True)
-
                 interface_dataframe.columns = ['uniprotID', 'positions']
-                st.write('HERE')
                 data = finalTouch(allData)
-                st.write('finalTouch')
-                st.write(data.head())
                 data = data.merge(interface_dataframe, on='uniprotID', how='left')
                 data.positions = data.positions.astype('str')
-                st.write('passed')
                 for i in data.index:
                     if (str(data.at[i, 'pos']) in data.at[i, 'positions']) and data.at[i, 'trsh4'] == 'surface':
                         data.at[i, 'threeState_trsh4_HQ'] = 'interface'
@@ -465,4 +459,6 @@ def pdb(input_set, mode, impute):
             minutes, seconds = divmod(rem, 60)
             print("Time passed: {:0>2}:{:0>2}:{:05.2f}".format(int(hours), int(minutes), seconds))
             #sys.stdout.close()
-            return data
+            return data
+    
+