Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -364,8 +364,6 @@ def alphafold(input_set, mode, impute):
                         except:
                             KeyError
                     info_per_model[mod][annot] = annotation_pos_on_pdb_
-                #pdb_path = Path(f'{alphafold_path}/AF-{uniprotID}-F{mod}-{file_str}.pdb.gz')
-
 
                 st.write('-----')
                 st.write('')
@@ -378,6 +376,7 @@ def alphafold(input_set, mode, impute):
                 with gzip.open(pdb_path, mode="rt") as f:
                     file_content = f.read()
                     st.write(file_content)
+                    break
                 st.write('REH')
                 st.write('-----')
                 st.write('')
@@ -389,13 +388,14 @@ def alphafold(input_set, mode, impute):
                 with gzip.open(pdb_path, mode="rt") as f:
                     file_content = f.read()
                     st.write(file_content)
+                    break
 
                 st.write('HERE2')
                 st.write('uniprotID',uniprotID)
                 st.write('model_num', mod)
                 st.write('pdb_path', pdb_path)
                 st.write('pdbSequence', pdbSequence)
-                st.write('mode',mode, type(mode))
+                st.write('mode',mode)
                 st.write(Path(path_to_output_files / '3D_alignment'))