Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -238,10 +238,12 @@ def pdb(input_set, mode, impute):
                 file = pdbl.retrieve_pdb_file(search, pdir=path_pdb, file_format="pdb")
                 st.write('file: ', file)
                 existing_pdb = list(Path(path_to_output_files / 'pdb_structures').glob("*"))
+                
                 st.write('after download:', existing_pdb)
                 st.write(Path(path_to_output_files / 'pdb_structures') , path_pdb)
-                existing_pdb = list(path_pdb.glob("*"))
-                st.write('after download:', existing_pdb)
+                
+                existing_pdb = list(glob.glob(f"{path_pdb}/*"))
+                st.write('after download 2:', existing_pdb)
 
                 resolution_method = parser.get_structure(search, file)
                 for record in SeqIO.parse(file, "pdb-seqres"):