Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -305,6 +305,8 @@ def pdb(input_set, mode, impute):
                 resolution_method = parser.get_structure(search, file)
                 for record in SeqIO.parse(file, "pdb-seqres"):
                     if record.dbxrefs[0].split(':')[0] == 'UNP':
+                        st.write(record.id.split(':')[0])
+                        st.write(record.id.split(':')[1])
                         pdb_fasta.at[index, 'pdbID'] = record.id.split(':')[0]
                         pdb_fasta.at[index, 'chain'] = record.id.split(':')[1]
                         pdb_fasta.at[index, 'pdbSequence'] = str(record.seq)