ficed bug in pdb parser

flatironinstitute · Sep 25, 2023 · ecf0020 · ecf0020
1 parent 29353ec
commit ecf0020
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cryo_sbi/utils/pdb_parser.py b/src/cryo_sbi/utils/pdb_parser.py
@@ -181,7 +181,7 @@ def pdb_parser(file_formatter, n_pdbs, output_file, mode, start_index=1):
     atomic_models = torch.zeros((n_pdbs, *atomic_model.shape))
 
     for i in range(0, n_pdbs):
-        atomic_models[i] = pdb_parser_(file_formatter.format(start_index+1), mode)
+        atomic_models[i] = pdb_parser_(file_formatter.format(start_index+i), mode)
 
 
     if output_file.endswith("pt"):