Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -232,8 +232,7 @@ def pdb(input_set, mode, impute):
                 response.raise_for_status()  # Check for a successful response
             except :
                 continue  # Skip to the next PDB code if fetching fails
-            st.write('response', response)
-            st.write(response.text[0:10])
+    
             # Step 2: Parse the PDB file from memory
             pdb_data = response.text
             pdb_parser = PDBParser(QUIET=True)  # QUIET=True suppresses warnings
@@ -249,7 +248,7 @@ def pdb(input_set, mode, impute):
                     seqs[i[2]] += i[4:]
                 else:
                     seqs[i[2]]  = i[4:]
-            from Bio.PDB.Polypeptide import *
+    
             for key, val in seqs.items():
                 seqs[key] = ''.join([threeToOne(i) for i in val])
             pdb_data_list = [i for i in pdb_data_list if i.startswith('DBREF')]